diff --git a/docker/Dockerfile b/docker/Dockerfile index 0faba17b9cee..49ff9b344268 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 .