Fix variable reference for CUDA version in setup script output messages

locaal-ai · Jan 31, 2025 · aaf798d · aaf798d
1 parent 6bf7605
commit aaf798d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup_cuda.ps1 b/setup_cuda.ps1
@@ -72,13 +72,13 @@ $cudaVersions = @{
 }
 
 if ($cudaVersions.ContainsKey($CUDA_VERSION_FULL)) {
-    $cudaInstallerUrl = $cudaVersions[$cudaVersion]
-    Write-Output "CUDA version $cudaVersion found. Downloading from $cudaInstallerUrl"
+    $cudaInstallerUrl = $cudaVersions[$CUDA_VERSION_FULL]
+    Write-Output "CUDA version $CUDA_VERSION_FULL found. Downloading from $cudaInstallerUrl"
     # Add your download and installation logic here
     $downloadUrl = $cudaInstallerUrl
 }
 else {
-    Write-Output "CUDA version $cudaVersion not found."
+    Write-Output "CUDA version $CUDA_VERSION_FULL not found."
     exit 1
 }