File tree Expand file tree Collapse file tree
aws/ami/windows/scripts/Installers Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ $windowsS3BaseUrl = "https://ossci-windows.s3.amazonaws.com"
1010$ProgressPreference = ' SilentlyContinue'
1111
1212# installerArgs
13- $installerArgs = " nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion nvprof_$cudaVersion cupti_$cudaVersion cublas_$cudaVersion cublas_dev_$cudaVersion cudart_$cudaVersion cufft_$cudaVersion cufft_dev_$cudaVersion curand_$cudaVersion curand_dev_$cudaVersion cusolver_$cudaVersion cusolver_dev_$cudaVersion cusparse_$cudaVersion cusparse_dev_$cudaVersion npp_$cudaVersion npp_dev_$cudaVersion nvrtc_$cudaVersion nvrtc_dev_$cudaVersion nvml_dev_$cudaVersion nvjpeg_$cudaVersion nvjpeg_dev_$cudaVersion "
13+ $installerArgs = " nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion nvprof_$cudaVersion cupti_$cudaVersion cublas_$cudaVersion cublas_dev_$cudaVersion cudart_$cudaVersion cufft_$cudaVersion cufft_dev_$cudaVersion curand_$cudaVersion curand_dev_$cudaVersion cusolver_$cudaVersion cusolver_dev_$cudaVersion cusparse_$cudaVersion cusparse_dev_$cudaVersion npp_$cudaVersion npp_dev_$cudaVersion nvrtc_$cudaVersion nvrtc_dev_$cudaVersion nvml_dev_$cudaVersion nvjpeg_$cudaVersion nvjpeg_dev_$cudaVersion cuda_profiler_api_ $cudaVersion nvjitlink_ $cudaVersion thrust_ $cudaVersion "
1414
1515# Switch statement for specfic CUDA versions
1616$cudnn_subfolder = " cuda"
@@ -21,26 +21,22 @@ $toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"
2121Switch ($cudaVersion ) {
2222 " 12.6" {
2323 $toolkitInstaller = " cuda_12.6.3_561.17_windows.exe"
24- $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion "
2524 }
2625 " 12.8" {
2726 $toolkitInstaller = " cuda_12.8.1_572.61_windows.exe"
28- $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion "
2927 }
3028 " 12.9" {
3129 $toolkitInstaller = " cuda_12.9.1_576.57_windows.exe"
32- $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion "
3330 }
3431 " 13.0" {
3532 $cudnn_subfolder = " cudnn-windows-x86_64-9.12.0.46_cuda13-archive"
3633 $toolkitInstaller = " cuda_13.0.0_windows.exe"
37- $installerArgs + = " cuda_profiler_api_ $cudaVersion nvjitlink_ $cudaVersion "
34+ $installerArgs = " "
3835 }
3936}
4037
4138
4239$cudnnZip = " $cudnn_subfolder .zip"
43- $installerArgs = " $installerArgs thrust_$cudaVersion "
4440$cudnn_lib_folder = " lib"
4541
4642Write-Output " Downloading ZLIB DLL, $windowsS3BaseUrl /zlib123dllx64.zip"
You can’t perform that action at this time.
0 commit comments