diff --git a/docker/Dockerfile.local b/docker/Dockerfile.local
index eef67412d5..7da00c0816 100644
--- 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
index f61c8515d4..8cd223aeb8 100644
--- 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