github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[docker] Add opencontainers image-spec to `Dockerfile` (#3006) 

* feat(docker): Add opencontainers image-spec to `Dockerfile`

This PR makes few changes to improve the overall quality of the docker image 🐳 . For reference more annotations can be found [here](https://github.com/opencontainers/image-spec/blob/main/annotations.md)

* feat(docker): add inline version declaration

* fix(docker): drop `org.opencontainers.image.version` LABEL
pull/3135/head
Saurav Maheshkar 2023-03-14 08:28:06 +00:00 committed by GitHub
parent 2eca4cd376
commit 1a46e71e07
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docker/Dockerfile
View File

@ -1,5 +1,10 @@
FROM hpcaitech/cuda-conda:11.3 FROM hpcaitech/cuda-conda:11.3
# metainformation
LABEL org.opencontainers.image.source = "https://github.com/hpcaitech/ColossalAI"
LABEL org.opencontainers.image.licenses = "Apache License 2.0"
LABEL org.opencontainers.image.base.name = "docker.io/library/hpcaitech/cuda-conda:11.3"
# install torch # install torch
RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
@ -22,4 +27,4 @@ RUN conda install cmake && \
git clone https://github.com/hpcaitech/TensorNVMe.git && \ git clone https://github.com/hpcaitech/TensorNVMe.git && \
cd TensorNVMe && \ cd TensorNVMe && \
pip install -r requirements.txt && \ pip install -r requirements.txt && \
pip install -v --no-cache-dir . pip install -v --no-cache-dir .