diff --git a/pytorch/cppbuild.sh b/pytorch/cppbuild.sh
index 16e3155849..617ee129c6 100755
--- a/pytorch/cppbuild.sh
+++ b/pytorch/cppbuild.sh
@@ -27,7 +27,7 @@ if [[ "$EXTENSION" == *gpu ]]; then
     export USE_CUDNN=1
     export USE_FAST_NVCC=0
     export CUDA_SEPARABLE_COMPILATION=OFF
-    export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;8.0+PTX"
+    export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;8.0;9.0"
 fi
 
 export PYTHON_BIN_PATH=$(which python3)