diff --git a/docker/Dockerfile b/docker/Dockerfile
index c6348364c..c3d7a7d21 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -138,7 +138,7 @@ RUN --mount=type=cache,target=/root/.cache/pip  python3 -m pip install --upgrade
 # We use Tom's DeepEP fork for GB200 for now; the 1fd57b0276311d035d16176bb0076426166e52f3 commit is https://github.com/fzyzcjy/DeepEP/tree/gb200_blog_part_2
 RUN set -eux; \
     if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
-      pip install nvidia-nvshmem-cu12==${NVSHMEM_VERSION} ; \
+      pip install nvidia-nvshmem-cu12==3.3.20 ; \
     elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \
       pip install nvidia-nvshmem-cu13==${NVSHMEM_VERSION} ; \
     else \
@@ -203,7 +203,7 @@ RUN --mount=type=cache,target=/root/.cache/pip python3 -m pip install \
 RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
     python3 -m pip install nvidia-nccl-cu12==2.28.3 --force-reinstall --no-deps ; \
     python3 -m pip install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall --no-deps; \
-    python3 -m pip install nvidia-nvshmem-cu12==${NVSHMEM_VERSION} --force-reinstall --no-deps; \
+    python3 -m pip install nvidia-nvshmem-cu12==3.3.20 --force-reinstall --no-deps; \
     python3 -m pip install nvidia-cutlass-dsl==4.3.0 --force-reinstall --no-deps; \
 elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \
     python3 -m pip install nvidia-nccl-cu13==2.28.3 --force-reinstall --no-deps ; \