From a3102e53589e37052dcf28663816ac643ecfe709 Mon Sep 17 00:00:00 2001 From: Samuel Audet Date: Tue, 31 Oct 2023 18:10:05 +0900 Subject: [PATCH] Try to update CUDA archs to "5.0;6.0;7.0;8.0+PTX" for PyTorch --- pytorch/cppbuild.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch/cppbuild.sh b/pytorch/cppbuild.sh index 29b7dabcf8..16e3155849 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+PTX" + export TORCH_CUDA_ARCH_LIST="5.0;6.0;7.0;8.0+PTX" fi export PYTHON_BIN_PATH=$(which python3)