2022-05-24 08:53:39 +00:00
|
|
|
FROM hpcaitech/pytorch-cuda:1.10.1-11.3.0
|
2022-01-07 06:54:04 +00:00
|
|
|
|
|
|
|
|
# install colossalai
|
|
|
|
|
RUN git clone https://github.com/hpcaitech/ColossalAI.git \
|
|
|
|
|
&& cd ./ColossalAI \
|
2022-02-14 09:09:30 +00:00
|
|
|
&& pip install -v --no-cache-dir .
|
