CUDA versioning switch case

atalman · Jul 12, 2022 · 3d7a662 · 3d7a662
1 parent 797a2f3
commit 3d7a662
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -28,11 +28,7 @@ Switch ($cudaVersion) {
     $toolkitInstaller = "cuda_10.2.89_441.22_win10.exe"
     $cudnnZip = "cudnn-10.2-windows10-x64-v7.6.5.32.zip"
   }
-  "11.1" {
-    $toolkitInstaller = "cuda_11.1.0_456.43_win10.exe"
-    $cudnnZip = "cudnn-11.1-windows-x64-v8.0.5.39.zip"
-  }
-  {($_ -eq "11.3") -or ($_ -eq "11.5") -or ($_ -eq "11.6")} {
+  {($_ -eq "11.3") -or ($_ -eq "11.6") -or ($_ -eq "11.7")} {
 
     Switch ($cudaVersion) {
       "11.3" {