From 1a46e71e077acc021a96bd3b25b8bd6013cd95e0 Mon Sep 17 00:00:00 2001 From: Saurav Maheshkar Date: Tue, 14 Mar 2023 08:28:06 +0000 Subject: [PATCH] [docker] Add opencontainers image-spec to `Dockerfile` (#3006) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * 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 --- docker/Dockerfile | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 0faba17b9..49ff9b344 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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 @@ -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 . \ No newline at end of file + pip install -v --no-cache-dir .