Add opencontainers image-spec to Dockerfile

DeepGraphLearning · Jul 4, 2022 · b0bd1c6 · b0bd1c6
1 parent e73d2c9
commit b0bd1c6
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,8 +1,15 @@
 FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
 
+# metainformation
+LABEL org.opencontainers.image.version = "0.1.3"
+LABEL org.opencontainers.image.authors = "TorchDrug Team"
+LABEL org.opencontainers.image.source = "https://github.com/DeepGraphLearning/torchdrug"
+LABEL org.opencontainers.image.licenses = "Apache License 2.0"
+LABEL org.opencontainers.image.base.name="docker.io/pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel"
+
 RUN apt-get update && \
     apt-get install -y libxrender1 && \
     rm -rf /var/lib/apt/lists/*
 
 RUN pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.1+cu111.html  && \
-    pip install torchdrug
+    pip install torchdrug