Skip to content

Commit a687b83

Browse files
authored
Install gpu driver DLLs for Windows CI (#2032)
1 parent 986d242 commit a687b83

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

packaging/windows/internal/cuda_install.bat

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,6 +96,11 @@ if not exist "%SRC_DIR%\temp_build\NvToolsExt.7z" (
9696
if errorlevel 1 exit /b 1
9797
)
9898

99+
if not exist "%SRC_DIR%\temp_build\gpu_driver_dlls.7z" (
100+
curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download" --output "%SRC_DIR%\temp_build\gpu_driver_dlls.7z"
101+
if errorlevel 1 exit /b 1
102+
)
103+
99104
echo Installing CUDA toolkit...
100105
7z x %CUDA_SETUP_FILE% -o"%SRC_DIR%\temp_build\cuda"
101106
pushd "%SRC_DIR%\temp_build\cuda"
@@ -131,5 +136,8 @@ xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\bin\*.*" "%ProgramFiles%\NVIDIA GPU Co
131136
xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\lib\x64\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\lib\x64"
132137
xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\include\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\include"
133138

139+
echo Installing GPU driver DLLs
140+
7z x %SRC_DIR%\temp_build\gpu_driver_dlls.zip -o"C:\Windows\System32"
141+
134142
echo Cleaning temp files
135143
rd /s /q "%SRC_DIR%\temp_build" || ver > nul

0 commit comments

Comments
 (0)