Skip to content

Commit

Permalink
.github/workflows/windows-cuda-dependency.yml: upgrade to cuda 12.6.0…
Browse files Browse the repository at this point in the history
… and tensorrt 10.4.0
  • Loading branch information
WolframRhodium committed Sep 7, 2024
1 parent c09faec commit 5ae95bc
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions .github/workflows/windows-cuda-dependency.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ jobs:
- name: Download TensorRT library
run: |
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.3.0/zip/TensorRT-10.3.0.26.Windows.win10.cuda-12.5.zip
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.4.0/zip/TensorRT-10.4.0.26.Windows.win10.cuda-12.6.zip
- name: Extract TensorRT library
run: |
Expand All @@ -55,14 +55,14 @@ jobs:
- name: Download CUDA Libraries
shell: cmd
run: |
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.5.1/network_installers/cuda_12.5.1_windows_network.exe
cuda_installer.exe -s cudart_12.5 cublas_12.5 cufft_12.5 cupti_12.5 nvrtc_12.5
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/12.6.0/network_installers/cuda_12.6.0_windows_network.exe
cuda_installer.exe -s cudart_12.6 cublas_12.6 cufft_12.6 cupti_12.6 nvrtc_12.6
- name: Move CUDA Libraries
shell: cmd
run: |
move "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5\extras\CUPTI\lib64\cupti*.dll" vsmlrt-cuda
move "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5\bin\*.dll" vsmlrt-cuda
move "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\extras\CUPTI\lib64\cupti*.dll" vsmlrt-cuda
move "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin\*.dll" vsmlrt-cuda
del vsmlrt-cuda\cudart32*.dll
- name: Setup VC commands
Expand Down

0 comments on commit 5ae95bc

Please sign in to comment.