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

update setup and workflow (#222)

Browse Source
This commit is contained in:
ver217
2022-02-14 17:09:30 +08:00
committed by Frank Lee
parent 3a1a9820b0
commit 578ea0583b
5 changed files with 159 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile
View File

@@ -8,4 +8,4 @@ RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple \
# install colossalai
RUN git clone https://github.com/hpcaitech/ColossalAI.git \
&& cd ./ColossalAI \
&& pip install -v --no-cache-dir --global-option="--cuda_ext" .
&& pip install -v --no-cache-dir .