github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added docker image (#126)

pull/129/head
Frank Lee 2022-01-07 14:54:04 +08:00 committed by GitHub
parent 293fb40c42
commit 42741dd4a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docker/Dockerfile Normal file
View File

@ -0,0 +1,10 @@
FROM nvcr.io/nvidia/pytorch:21.07-py3
# install dependencies
RUN 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 --global-option="--cuda_ext" .