Skip to content

Commit

Permalink
Update install-cuda.ps1
Browse files Browse the repository at this point in the history
  • Loading branch information
jrhemstad authored Oct 25, 2023
1 parent d25fe6a commit 0d8b57b
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions windows/image/installers/install-cuda.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,8 @@ $cudaUri = @{
"12.1" = "12.1.1/network_installers/cuda_12.1.1_windows_network.exe"
"12.2" = "12.2.0/network_installers/cuda_12.2.0_windows_network.exe"
"12.2.0" = "12.2.0/network_installers/cuda_12.2.0_windows_network.exe"
"latest" = "12.2.0/network_installers/cuda_12.2.0_windows_network.exe"
"12.3.0" = "12.3.0/network_installers/cuda_12.2.0_windows_network.exe"
"latest" = "12.3.0/network_installers/cuda_12.2.0_windows_network.exe"
}[$cudaVersion]

# The component tags don't include patch, so we need to map input to major.minor
Expand All @@ -33,7 +34,9 @@ $componentTag = @{
"12.1" = "12.1"
"12.2" = "12.2"
"12.2.0" = "12.2"
"latest" = "12.2"
"12.3" = "12.3"
"12.3.0" = "12.3"
"latest" = "12.3"
}[$cudaVersion]

$cudaVersionUrl = "https://developer.download.nvidia.com/compute/cuda/$cudaUri"
Expand Down

0 comments on commit 0d8b57b

Please sign in to comment.