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

bytedeco · Oct 31, 2023 · a3102e5 · a3102e5
1 parent 41a97d1
commit a3102e5
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+PTX"
+    export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;8.0+PTX"
 fi
 
 export PYTHON_BIN_PATH=$(which python3)