@@ -56,18 +56,18 @@ if [[ -n "$build_with_cuda" ]]; then
56
56
export USE_STATIC_CUDNN=1 # links cudnn statically (driven by tools/setup_helpers/cudnn.py)
57
57
58
58
if [[ $CUDA_VERSION == 11.8* ]]; then
59
- TORCH_CUDA_ARCH_LIST=" $TORCH_CUDA_ARCH_LIST ;3.7+PTX;9.0"
60
- # for cuda 11.8 we use cudnn 8.7
61
- # which does not have single static libcudnn_static.a deliverable to link with
62
- export USE_STATIC_CUDNN=0
63
- # for cuda 11.8 include all dynamic loading libraries
64
- DEPS_LIST=(/usr/local/cuda/lib64/libcudnn* .so.8 /usr/local/cuda-11.8/extras/CUPTI/lib64/libcupti.so.11.8)
59
+ TORCH_CUDA_ARCH_LIST=" $TORCH_CUDA_ARCH_LIST ;3.7+PTX;9.0"
60
+ # for cuda 11.8 we use cudnn 8.7
61
+ # which does not have single static libcudnn_static.a deliverable to link with
62
+ export USE_STATIC_CUDNN=0
63
+ # for cuda 11.8 include all dynamic loading libraries
64
+ DEPS_LIST=(/usr/local/cuda/lib64/libcudnn* .so.8 /usr/local/cuda-11.8/extras/CUPTI/lib64/libcupti.so.11.8)
65
65
elif [[ $CUDA_VERSION == 12.1* ]]; then
66
- # cuda 12 does not support sm_3x
67
- TORCH_CUDA_ARCH_LIST=" $TORCH_CUDA_ARCH_LIST ;9.0"
68
- # for cuda 12.1 we use cudnn 8.8 and include all dynamic loading libraries
69
- export USE_STATIC_CUDNN=0
70
- DEPS_LIST=(/usr/local/cuda/lib64/libcudnn* .so.8 /usr/local/cuda-12.1/extras/CUPTI/lib64/libcupti.so.12)
66
+ # cuda 12 does not support sm_3x
67
+ TORCH_CUDA_ARCH_LIST=" $TORCH_CUDA_ARCH_LIST ;9.0"
68
+ # for cuda 12.1 we use cudnn 8.8 and include all dynamic loading libraries
69
+ export USE_STATIC_CUDNN=0
70
+ DEPS_LIST=(/usr/local/cuda/lib64/libcudnn* .so.8 /usr/local/cuda-12.1/extras/CUPTI/lib64/libcupti.so.12)
71
71
fi
72
72
if [[ -n " $OVERRIDE_TORCH_CUDA_ARCH_LIST " ]]; then
73
73
TORCH_CUDA_ARCH_LIST=" $OVERRIDE_TORCH_CUDA_ARCH_LIST "
0 commit comments