DGL Docker (#551)

Добавил Docker для DGL. Проверил в gitHub Workspace - всё работает.
itlab-vision · Oct 7, 2024 · f8994ef · f8994ef
1 parent b0d60e6
commit f8994ef
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/docker/DGL/Dockerfile b/docker/DGL/Dockerfile
@@ -0,0 +1,22 @@
+FROM ubuntu_for_dli
+
+WORKDIR /root/
+
+ARG TORCH_VERSION=2.2.1
+ARG TORCHAUDIO_VERSION=2.2.1
+ARG TORCHVISION_VERSION=0.17.1
+
+# Update pip
+WORKDIR /tmp/
+RUN pip3 install --upgrade pip
+
+# Install PyTorch Python
+RUN pip3 install torch==${TORCH_VERSION} \
+                 torchvision==${TORCHVISION_VERSION} \
+                 torchaudio==${TORCHAUDIO_VERSION}
+
+# Install DGL(PyTorch)
+RUN pip3 install dgl==2.0.0 -f https://data.dgl.ai/wheels/repo.html
+RUN pip3 install dglgo==0.0.2 -f https://data.dgl.ai/wheels-test/repo.html
+
+WORKDIR /tmp/