Skip to content

Commit

Permalink
[docker] Add opencontainers image-spec to Dockerfile (hpcaitech#3006)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
SauravMaheshkar authored Mar 14, 2023
1 parent 2eca4cd commit 1a46e71
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
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
RUN conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

Expand All @@ -22,4 +27,4 @@ RUN conda install cmake && \
git clone https://github.com/hpcaitech/TensorNVMe.git && \
cd TensorNVMe && \
pip install -r requirements.txt && \
pip install -v --no-cache-dir .
pip install -v --no-cache-dir .

0 comments on commit 1a46e71

Please sign in to comment.