From 49668cbf900ee81571f4c2f1781c1ad3d94bf6d4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20Guillemet?= Date: Wed, 1 Nov 2023 09:11:30 +0100 Subject: [PATCH] Try to update CUDA archs to "5.0;6.0;7.0;8.0;9.0" for PyTorch --- pytorch/cppbuild.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch/cppbuild.sh b/pytorch/cppbuild.sh index 235a554be01..8e9d089548b 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)