diff --git a/windows/image/installers/install-cuda.ps1 b/windows/image/installers/install-cuda.ps1 index 3cbb04c1..1c2f34c9 100644 --- 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"