mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-16 07:28:45 +00:00
7 lines
182 B
Docker
7 lines
182 B
Docker
FROM hpcaitech/pytorch-cuda:1.10.1-11.3.0
|
|
|
|
# install colossalai
|
|
RUN git clone https://github.com/hpcaitech/ColossalAI.git \
|
|
&& cd ./ColossalAI \
|
|
&& pip install -v --no-cache-dir .
|