diff --git a/.jenkins/pytorch/build.sh b/.jenkins/pytorch/build.sh
index e6b96605d56..2f37a51c105 100755
--- a/.jenkins/pytorch/build.sh
+++ b/.jenkins/pytorch/build.sh
@@ -184,8 +184,8 @@ fi
 # Target only our CI GPU machine's CUDA arch to speed up the build
 export TORCH_CUDA_ARCH_LIST="5.2"
 
-# Add sm_75 support for the Linux CUDA 10.2 cuDNN 7 build
-if [[ "$BUILD_ENVIRONMENT" == *cuda10.2-cudnn7*build ]]; then
+# Add sm_75 support for the Linux CUDA 11.1 cuDNN 8 CircleCI build
+if [[ "$BUILD_ENVIRONMENT" == *xenial-cuda11.1-cudnn8*build ]]; then
   export TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST";7.5"
 fi