mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
373 B
11 lines
373 B
FROM nvcr.io/nvidia/pytorch:21.07-py3 |
|
|
|
# install dependencies |
|
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \ |
|
&& pip install -U pip setuptools wheel \ |
|
&& pip install pytest tensorboard deepspeed apex |
|
|
|
# install colossalai |
|
RUN git clone https://github.com/hpcaitech/ColossalAI.git \ |
|
&& cd ./ColossalAI \ |
|
&& pip install -v --no-cache-dir .
|
|
|