Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
Dockerfile 182 Bytes
Newer Older
Frank Lee's avatar
[ci] update the docker image name (#1017)  
Frank Lee committed
1 
FROM hpcaitech/pytorch-cuda:1.10.1-11.3.0
Frank Lee's avatar
added docker image (#126)  
Frank Lee committed
2
3
4
5 

# install colossalai
RUN git clone https://github.com/hpcaitech/ColossalAI.git \
 && cd ./ColossalAI \
ver217's avatar
update setup and workflow (#222)  
ver217 committed
6 
 && pip install -v --no-cache-dir .