Add NVTX to Windows build. (rapidsai#277)

msarahan · Apr 25, 2024 · 7013588 · 7013588
1 parent 069579f
commit 7013588
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/windows/image/installers/install-cuda.ps1 b/windows/image/installers/install-cuda.ps1
@@ -59,7 +59,8 @@ $cudaComponents =
     "cupti_$componentTag",
     "nvrtc_$componentTag",
     "nvrtc_dev_$componentTag",
-    "nvml_dev_$componentTag"
+    "nvml_dev_$componentTag",
+    "nvtx_$componentTag"
 
 Invoke-WebRequest -Uri "$cudaVersionUrl" -OutFile "./cuda_network.exe" -UseBasicParsing
 Start-Process -Wait -PassThru -FilePath .\cuda_network.exe -ArgumentList "-s $cudaComponents"