File tree Expand file tree Collapse file tree 1 file changed +1
-10
lines changed Expand file tree Collapse file tree 1 file changed +1
-10
lines changed Original file line number Diff line number Diff line change @@ -64,7 +64,7 @@ echo "install conda package"
64
64
set " CONDA_HOME = %CD% \conda"
65
65
set " tmp_conda = %CONDA_HOME% "
66
66
set " miniconda_exe = %CD% \miniconda.exe"
67
- set " CONDA_EXTRA_ARGS = "
67
+ set " CONDA_EXTRA_ARGS = cpuonly -c pytorch-nightly "
68
68
if " %CUDA_VERSION% " == " 116" (
69
69
set " CONDA_EXTRA_ARGS = pytorch-cuda=11.6 -c nvidia -c pytorch-nightly"
70
70
)
@@ -102,15 +102,6 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
102
102
for /F " delims=" %%i in ('where /R " %PYTORCH_FINAL_PACKAGE_DIR:/ =\ % " *.tar.bz2') do call conda install -yq " %%i " --offline
103
103
if ERRORLEVEL 1 exit /b 1
104
104
105
- if " %CUDA_VERSION% " == " cpu" goto install_cpu_torch
106
-
107
- goto smoke_test
108
-
109
- :install_cpu_torch
110
- call conda install %CONDA_EXTRA_ARGS% -y cpuonly -c pytorch-nightly
111
- if ERRORLEVEL 1 exit /b 1
112
-
113
- :smoke_test
114
105
python -c " import torch"
115
106
if ERRORLEVEL 1 exit /b 1
116
107
You can’t perform that action at this time.
0 commit comments