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

[workflow] auto-publish docker image upon release (#1164)

Browse Source
This commit is contained in:
Frank Lee
2022-06-23 14:51:59 +08:00
committed by GitHub
parent 634eecb98e
commit ca73028a3a
3 changed files with 50 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docker/Dockerfile
View File

@@ -1,4 +1,12 @@
FROM hpcaitech/pytorch-cuda:1.10.1-11.3.0
FROM hpcaitech/cuda-conda:11.3
# install torch
RUN conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
# install apex
RUN git clone https://github.com/NVIDIA/apex && \
cd apex && \
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
# install colossalai
RUN git clone https://github.com/hpcaitech/ColossalAI.git \