github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 17:46:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

added docker image (#126)

Browse Source
This commit is contained in:
Frank Lee
2022-01-07 14:54:04 +08:00
committed by GitHub
parent 293fb40c42
commit 42741dd4a3
1 changed files with 10 additions and 0 deletions
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" .