Add NVML to the Windows CUDA components list. (#260)

* Add NVML to the CUDA components list. Also moves the array of CUDA components somewhere more accessible to edit. * Add cupti to the WIndows CUDA installer.
rapidsai · Apr 8, 2024 · e42ef92 · e42ef92
1 parent d7ae3d4
commit e42ef92
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/windows/image/installers/install-cuda.ps1 b/windows/image/installers/install-cuda.ps1
@@ -51,9 +51,18 @@ $componentTag = @{
 }[$cudaVersion]
 
 $cudaVersionUrl = "https://developer.download.nvidia.com/compute/cuda/$cudaUri"
-
+$cudaComponents = 
+    "nvcc_$componentTag",
+    "curand_$componentTag",
+    "curand_dev_$componentTag",
+    "cudart_$componentTag",
+    "cupti_$componentTag",
+    "nvrtc_$componentTag",
+    "nvrtc_dev_$componentTag",
+    "nvml_dev_$componentTag"
+
 Invoke-WebRequest -Uri "$cudaVersionUrl" -OutFile "./cuda_network.exe" -UseBasicParsing
-Start-Process -Wait -PassThru -FilePath .\cuda_network.exe -ArgumentList "-s nvcc_$componentTag curand_$componentTag curand_dev_$componentTag cudart_$componentTag nvrtc_$componentTag nvrtc_dev_$componentTag"
+Start-Process -Wait -PassThru -FilePath .\cuda_network.exe -ArgumentList "-s $cudaComponents"
 
 $ENV:PATH="$ENV:PATH;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v$componentTag\bin"
 $ENV:CUDA_PATH="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v$componentTag"