Skip to content

Commit 377713a

Browse files
authored
add CUDA VS intregration install to Windows AMI build script (#4189)
Follow-up to pytorch/vision#7417 (comment). Untested, but should work.
1 parent 3382612 commit 377713a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@ function Install-CudaToolkit() {
8080
Get-Content -Path "$cudaInstallLogs\LOG.setup.exe.log"
8181
exit 1
8282
}
83+
84+
Write-Output "Installing VS 2019 integration"
85+
Copy-Item -Force -Verbose -Recurse "$tmpExtractedInstaller\visual_studio_integration\CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions\*.*" "C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools\MSBuild\Microsoft\VC\v160\BuildCustomizations"
8386
}
8487

8588
function Install-Cudnn() {

0 commit comments

Comments
 (0)