frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't link with libnvToolsExt when building for 12.9 (#165465)

Browse Source 
This is to bring back this logic from https://github.com/pytorch/pytorch/pull/161916/files#diff-bf46b4a09ca67e50622bf84fefc0d11b584ffcc24ee6cc5019cf0fc7565d81a8L170.  Building libtorch on 12.9 is failing otherwise https://github.com/pytorch/pytorch/actions/runs/18458531395/job/52610761895:

```
cp: cannot stat '/usr/local/cuda/lib64/libnvToolsExt.so.1': No such file or directory
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165465
Approved by: https://github.com/atalman, https://github.com/malfet
This commit is contained in:
Huy Do
2025-10-15 01:45:37 +00:00
committed by PyTorch MergeBot
parent a20afb6100
commit 132ae8e6dd
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
.ci/manywheel/build_cuda.sh
View File

@ -187,19 +187,22 @@ if [[ $CUDA_VERSION == 12* || $CUDA_VERSION == 13* ]]; then
export USE_CUFILE=0 export USE_CUFILE=0
else else
DEPS_LIST+=( DEPS_LIST+=(
"/usr/local/cuda/lib64/libnvToolsExt.so.1"
"/usr/local/cuda/lib64/libcublas.so.12" "/usr/local/cuda/lib64/libcublas.so.12"
"/usr/local/cuda/lib64/libcublasLt.so.12" "/usr/local/cuda/lib64/libcublasLt.so.12"
"/usr/local/cuda/lib64/libcudart.so.12" "/usr/local/cuda/lib64/libcudart.so.12"
"/usr/local/cuda/lib64/libnvrtc.so.12" "/usr/local/cuda/lib64/libnvrtc.so.12"
"/usr/local/cuda/extras/CUPTI/lib64/libcupti.so.12") "/usr/local/cuda/extras/CUPTI/lib64/libcupti.so.12")
DEPS_SONAME+=( DEPS_SONAME+=(
"libnvToolsExt.so.1"
"libcublas.so.12" "libcublas.so.12"
"libcublasLt.so.12" "libcublasLt.so.12"
"libcudart.so.12" "libcudart.so.12"
"libnvrtc.so.12" "libnvrtc.so.12"
"libcupti.so.12") "libcupti.so.12")
if [[ $CUDA_VERSION != 12.9* ]]; then
DEPS_LIST+=("/usr/local/cuda/lib64/libnvToolsExt.so.1")
DEPS_SONAME+=("libnvToolsExt.so.1")
fi
fi fi
else else
echo "Using nvidia libs from pypi." echo "Using nvidia libs from pypi."