"vscode:/vscode.git/clone" did not exist on "943982d29a2a6bd6c74012e6795fcf9af7a63ea5"
Dockerfile 373 Bytes
Newer Older
Frank Lee's avatar
Frank Lee committed
1
2
3
FROM nvcr.io/nvidia/pytorch:21.07-py3

# install dependencies
Frank Lee's avatar
Frank Lee committed
4
5
RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \
 && pip install -U pip setuptools wheel \
Frank Lee's avatar
Frank Lee committed
6
7
8
9
10
 && pip install pytest tensorboard deepspeed apex

# install colossalai
RUN git clone https://github.com/hpcaitech/ColossalAI.git \
 && cd ./ColossalAI \
ver217's avatar
ver217 committed
11
 && pip install -v --no-cache-dir .