added docker documentation (#152)

This commit is contained in:
Frank Lee
2022-01-18 13:35:18 +08:00
committed by GitHub
parent 7bf1e98b97
commit a1da3900c8
2 changed files with 18 additions and 1 deletions

View File

@@ -1,7 +1,8 @@
FROM nvcr.io/nvidia/pytorch:21.07-py3
# install dependencies
RUN pip install -U pip setuptools wheel \
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