Try to update CUDA archs to "5.0;6.0;7.0;8.0;9.0" for PyTorch

bytedeco · Nov 1, 2023 · 9e08093 · 9e08093
1 parent a3102e5
commit 9e08093
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)