Update the docker compose file to use the reference implementation in…

… the training repo instead of gltorch
mlcommons · Apr 26, 2024 · 3a48931 · 3a48931
1 parent 5c08ce5
commit 3a48931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graph_neural_network/Dockerfile b/graph_neural_network/Dockerfile
@@ -15,5 +15,5 @@ RUN pip install git+https://github.com/mlcommons/logging.git
 # TF32 instead of FP32 for faster compute
 ENV NVIDIA_TF32_OVERRIDE=1
 
-RUN git clone https://github.com/alibaba/graphlearn-for-pytorch.git
-WORKDIR /workspace/repository/graphlearn-for-pytorch/examples/igbh
+COPY ..
+WORKDIR /workspace/repository