Try removing super old cuda arch
if [[ "$cuda_compiler_version" == "None" ]]; then
export THESEUS_FORCE_CUDA=0
else
- export TORCH_CUDA_ARCH_LIST="3.5;5.0"
if [[ ${cuda_compiler_version} == 9.0* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;7.0"
+ export TORCH_CUDA_ARCH_LIST="6.0;7.0"
elif [[ ${cuda_compiler_version} == 9.2* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0"
+ export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0"
elif [[ ${cuda_compiler_version} == 10.* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5"
+ export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5"
elif [[ ${cuda_compiler_version} == 11.0* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0"
+ export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0"
elif [[ ${cuda_compiler_version} == 11.1* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
+ export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6"
elif [[ ${cuda_compiler_version} == 11.2* ]]; then
- export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
+ export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6"
else
echo "unsupported cuda version. edit build.sh"
exit 1
