From feb02013eada5d46271656caba55f3a0447e77bf Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Thu, 21 Oct 2021 20:09:34 -0700 Subject: [PATCH] Do not install Display.Driver on GPU-less machines GPU tests have special step to install the driver, CUDA install step shouldl just install the runtime never Fixes regression introduced by https://github.com/pytorch/vision/pull/4421 --- packaging/windows/internal/cuda_install.bat | 4 ---- 1 file changed, 4 deletions(-) diff --git a/packaging/windows/internal/cuda_install.bat b/packaging/windows/internal/cuda_install.bat index db0e19320fb..e80654077ca 100644 --- a/packaging/windows/internal/cuda_install.bat +++ b/packaging/windows/internal/cuda_install.bat @@ -135,10 +135,6 @@ if not exist "%SRC_DIR%\temp_build\cuda_11.1.1_456.81_win10.exe" ( set "ARGS=nvcc_11.1 cuobjdump_11.1 nvprune_11.1 nvprof_11.1 cupti_11.1 cublas_11.1 cublas_dev_11.1 cudart_11.1 cufft_11.1 cufft_dev_11.1 curand_11.1 curand_dev_11.1 cusolver_11.1 cusolver_dev_11.1 cusparse_11.1 cusparse_dev_11.1 npp_11.1 npp_dev_11.1 nvjpeg_11.1 nvjpeg_dev_11.1 nvrtc_11.1 nvrtc_dev_11.1 nvml_dev_11.1" ) -@REM There is no downloadable driver for Tesla on CUDA 11.1 yet. We will use -@REM the driver inside CUDA -set "ARGS=%ARGS% Display.Driver" - if not exist "%SRC_DIR%\temp_build\cudnn-11.1-windows-x64-v8.0.5.39.zip" ( curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-11.1-windows-x64-v8.0.5.39.zip --output "%SRC_DIR%\temp_build\cudnn-11.1-windows-x64-v8.0.5.39.zip" if errorlevel 1 exit /b 1