Skip to content

Commit

Permalink
Update cudnn to 8.9.7.29 to align with pytorch/pytorch#120642
Browse files Browse the repository at this point in the history
  • Loading branch information
nWEIdia committed Mar 12, 2024
1 parent e63ead5 commit 978978b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion windows/internal/cuda_install.bat
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
set "ARGS=cuda_profiler_api_12.4 thrust_12.4 nvcc_12.4 cuobjdump_12.4 nvprune_12.4 nvprof_12.4 cupti_12.4 cublas_12.4 cublas_dev_12.4 cudart_12.4 cufft_12.4 cufft_dev_12.4 curand_12.4 curand_dev_12.4 cusolver_12.4 cusolver_dev_12.4 cusparse_12.4 cusparse_dev_12.4 npp_12.4 npp_dev_12.4 nvrtc_12.4 nvrtc_dev_12.4 nvml_dev_12.4 nvjitlink_12.4 nvtx_12.4"
)

set CUDNN_FOLDER=cudnn-windows-x86_64-9.0.0.312_cuda12-archive
set CUDNN_FOLDER=cudnn-windows-x86_64-8.9.7.29_cuda12-archive
set CUDNN_LIB_FOLDER="lib"
set "CUDNN_INSTALL_ZIP=%CUDNN_FOLDER%.zip"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
Expand Down

0 comments on commit 978978b

Please sign in to comment.