From 9f42bc75cbf8f9015e8831cf27d3f35de6d489a0 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 26 Oct 2021 03:51:17 -0700 Subject: [PATCH] [fbsync] Do not install Display.Driver on GPU-less machines (#4695) Summary: 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 Reviewed By: NicolasHug Differential Revision: D31916340 fbshipit-source-id: 700b522fcfe58942603f8eb336811c585c28330e --- 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