Try to increase TORCH_CUDA_ARCH_LIST to "5.0;6.0;7.0+PTX"

bytedeco · Jun 13, 2023 · 21fb4b9 · 21fb4b9
1 parent b0da0db
commit 21fb4b9
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+PTX"
+    export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0+PTX"
 fi
 
 export PYTHON_BIN_PATH=$(which python3)