File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ export USE_CUDA_STATIC_LINK=1
14
14
. ./switch_cuda_version.sh 9.0
15
15
16
16
17
- conda install -y cmake=3.22.3 numpy=1.17 setuptools pyyaml cffi mkl=2018 mkl-include typing_extension ninja magma-cuda80 -c pytorch
17
+ conda install -y cmake numpy=1.17 setuptools pyyaml cffi mkl=2018 mkl-include typing_extension ninja magma-cuda80 -c pytorch
18
18
19
19
export CMAKE_PREFIX_PATH=" $( dirname $( which conda) ) /../"
20
20
git clone https://github.com/pytorch/pytorch -b nightly2 --recursive
Original file line number Diff line number Diff line change @@ -166,7 +166,7 @@ tmp_env_name="wheel_py$python_nodot"
166
166
conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn " $tmp_env_name " python=" $desired_python "
167
167
source activate " $tmp_env_name "
168
168
169
- retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake=3.22.3 " numpy${NUMPY_PINNED_VERSION} " nomkl " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " cffi typing_extensions ninja requests
169
+ retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake " numpy${NUMPY_PINNED_VERSION} " nomkl " setuptools${SETUPTOOLS_PINNED_VERSION} " " pyyaml${PYYAML_PINNED_VERSION} " cffi typing_extensions ninja requests
170
170
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2020.1 mkl-static==2020.1 -c intel
171
171
retry pip install -qr " ${pytorch_rootdir} /requirements.txt" || true
172
172
You can’t perform that action at this time.
0 commit comments