Update dgl version. (#271)

*Issue #, if available:* *Description of changes:* By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. --------- Co-authored-by: Xiang Song <[email protected]>
awslabs · Jun 23, 2023 · eb01fb7 · eb01fb7
1 parent f7393ea
commit eb01fb7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docker/Dockerfile.local b/docker/Dockerfile.local
@@ -12,7 +12,7 @@ RUN apt-get install -y cmake
 RUN pip install torch==1.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
 
 # Install DGL
-RUN pip3 install dgl==1.0.3 -f https://data.dgl.ai/wheels/cu116/repo.html
+RUN pip3 install dgl==1.0.3+cu117 -f https://data.dgl.ai/wheels/cu117/repo.html
 
 # Install related Python packages
 RUN pip3 install ogb==1.3.6 scipy pyarrow boto3 scikit-learn transformers

diff --git a/docker/sagemaker/Dockerfile.sm b/docker/sagemaker/Dockerfile.sm
@@ -8,7 +8,7 @@ FROM 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.13.1-gpu-py
 LABEL maintainer="Amazon AI Graph ML team"
 
 # Install DGL
-RUN pip3 install dgl==1.0.3 -f https://data.dgl.ai/wheels/cu117/repo.html
+RUN pip3 install dgl==1.0.3+cu117 -f https://data.dgl.ai/wheels/cu117/repo.html
 
 # Install related Python packages
 RUN pip3 install ogb==1.3.6 scipy pyarrow boto3 scikit-learn transformers