Update windows smoke test for CUDA 11.1 (#629)

pytorch · Jan 29, 2021 · a4867f4 · a4867f4
1 parent e843eaf
commit a4867f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/windows/internal/smoke_test.bat b/windows/internal/smoke_test.bat
@@ -4,6 +4,7 @@ pushd %SRC_DIR%\..
 
 if "%CUDA_VERSION%" == "102" call internal\driver_update.bat
 if "%CUDA_VERSION%" == "110" call internal\driver_update.bat
+if "%CUDA_VERSION%" == "111" call internal\driver_update.bat
 if errorlevel 1 exit /b 1
 
 set "ORIG_PATH=%PATH%"