From 3d7a6625f0480b6626e14febb2d9e96df59a7164 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Tue, 12 Jul 2022 05:52:44 -0700
Subject: [PATCH] CUDA versioning switch case

---
 aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
index fa7acd49e5..ff602c3a96 100644
--- 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" {