File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
.ci/pytorch/win-test-helpers Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -37,6 +37,11 @@ call %INSTALLER_DIR%\activate_miniconda3.bat
37
37
if errorlevel 1 goto fail
38
38
if not errorlevel 0 goto fail
39
39
40
+ :: Update CMake
41
+ call choco upgrade -y cmake --no-progress --installargs 'ADD_CMAKE_TO_PATH=System' --apply-install-arguments-to-dependencies --version=3.27.9
42
+ if errorlevel 1 goto fail
43
+ if not errorlevel 0 goto fail
44
+
40
45
call pip install mkl-include== 2021.4.0 mkl-devel== 2021.4.0
41
46
if errorlevel 1 goto fail
42
47
if not errorlevel 0 goto fail
@@ -88,7 +93,7 @@ set PATH=%CUDA_PATH%\bin;%CUDA_PATH%\libnvvp;%PATH%
88
93
:cuda_build_end
89
94
90
95
set DISTUTILS_USE_SDK = 1
91
- set PATH = %TMP_DIR_WIN% \bin;%PATH%
96
+ set PATH = %TMP_DIR_WIN% \bin;C:\Program Files\CMake\bin; %PATH%
92
97
93
98
:: The latest Windows CUDA test is running on AWS G5 runner with A10G GPU
94
99
if " %TORCH_CUDA_ARCH_LIST% " == " " set TORCH_CUDA_ARCH_LIST = 8.6
You can’t perform that action at this time.
0 commit commentsComments 0 (0)