Skip to content

Commit bc4176c

Browse files
atalmanpytorchmergebot
authored andcommitted
CD Windows CUDA 13.0 build - fix packaging of cuda dlls (pytorch#162383)
Trying to fix pytorch#162333 CUDA 13.0 file structure changed. Instead of keeping most of dlls in bin folder its now in ``bin\x64`` except for cudnn dll. See attached picture : <img width="511" height="361" alt="Screenshot 2025-09-08 at 9 46 26 AM" src="https://github.com/user-attachments/assets/d2e630ee-930f-4da6-9b81-f9ef48fde7ce" /> <img width="490" height="333" alt="Screenshot 2025-09-08 at 9 46 34 AM" src="https://github.com/user-attachments/assets/194cbf43-b6ef-4218-b516-db37b91302be" /> Pull Request resolved: pytorch#162383 Approved by: https://github.com/seemethere, https://github.com/ZainRizvi, https://github.com/malfet
1 parent de5dc1f commit bc4176c

File tree

1 file changed

+15
-12
lines changed

1 file changed

+15
-12
lines changed

.ci/pytorch/windows/internal/copy.bat

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,20 @@
1-
copy "%CUDA_PATH%\bin\cusparse*64_*.dll*" pytorch\torch\lib
2-
copy "%CUDA_PATH%\bin\cublas*64_*.dll*" pytorch\torch\lib
3-
copy "%CUDA_PATH%\bin\cudart*64_*.dll*" pytorch\torch\lib
4-
copy "%CUDA_PATH%\bin\curand*64_*.dll*" pytorch\torch\lib
5-
copy "%CUDA_PATH%\bin\cufft*64_*.dll*" pytorch\torch\lib
6-
copy "%CUDA_PATH%\bin\cusolver*64_*.dll*" pytorch\torch\lib
1+
2+
if %CUDA_VERSION% geq 130 (
3+
set "dll_path=bin\x64"
4+
) else (
5+
set "dll_path=bin"
6+
)
7+
8+
copy "%CUDA_PATH%\%dll_path%\cusparse*64_*.dll*" pytorch\torch\lib
9+
copy "%CUDA_PATH%\%dll_path%\cublas*64_*.dll*" pytorch\torch\lib
10+
copy "%CUDA_PATH%\%dll_path%\cudart*64_*.dll*" pytorch\torch\lib
11+
copy "%CUDA_PATH%\%dll_path%\curand*64_*.dll*" pytorch\torch\lib
12+
copy "%CUDA_PATH%\%dll_path%\cufft*64_*.dll*" pytorch\torch\lib
13+
copy "%CUDA_PATH%\%dll_path%\cusolver*64_*.dll*" pytorch\torch\lib
14+
copy "%CUDA_PATH%\%dll_path%\nvrtc*64_*.dll*" pytorch\torch\lib
15+
copy "%CUDA_PATH%\%dll_path%\nvJitLink_*.dll*" pytorch\torch\lib
716

817
copy "%CUDA_PATH%\bin\cudnn*64_*.dll*" pytorch\torch\lib
9-
copy "%CUDA_PATH%\bin\nvrtc*64_*.dll*" pytorch\torch\lib
1018
copy "%CUDA_PATH%\extras\CUPTI\lib64\cupti64_*.dll*" pytorch\torch\lib
1119
copy "%CUDA_PATH%\extras\CUPTI\lib64\nvperf_host*.dll*" pytorch\torch\lib
1220

@@ -20,8 +28,3 @@ copy "%libuv_ROOT%\bin\uv.dll" pytorch\torch\lib
2028
if exist "C:\Windows\System32\zlibwapi.dll" (
2129
copy "C:\Windows\System32\zlibwapi.dll" pytorch\torch\lib
2230
)
23-
24-
::copy nvJitLink dll is requires for cuda 12+
25-
if exist "%CUDA_PATH%\bin\nvJitLink_*.dll*" (
26-
copy "%CUDA_PATH%\bin\nvJitLink_*.dll*" pytorch\torch\lib
27-
)

0 commit comments

Comments
 (0)