Skip to content

Commit 3a48931

Browse files
committed
Update the docker compose file to use the reference implementation in the training repo instead of gltorch
1 parent 5c08ce5 commit 3a48931

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graph_neural_network/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,5 @@ RUN pip install git+https://github.com/mlcommons/logging.git
1515
# TF32 instead of FP32 for faster compute
1616
ENV NVIDIA_TF32_OVERRIDE=1
1717

18-
RUN git clone https://github.com/alibaba/graphlearn-for-pytorch.git
19-
WORKDIR /workspace/repository/graphlearn-for-pytorch/examples/igbh
18+
COPY ..
19+
WORKDIR /workspace/repository

0 commit comments

Comments
 (0)