diff --git a/conda/debugging_pytorch.sh b/conda/debugging_pytorch.sh index 2de2bef0b..5fc964986 100644 --- a/conda/debugging_pytorch.sh +++ b/conda/debugging_pytorch.sh @@ -14,7 +14,7 @@ export USE_CUDA_STATIC_LINK=1 . ./switch_cuda_version.sh 9.0 -conda install -y cmake numpy=1.11 setuptools pyyaml cffi mkl=2018 mkl-include typing ninja magma-cuda80 -c pytorch +conda install -y cmake numpy=1.17 setuptools pyyaml cffi mkl=2018 mkl-include typing ninja magma-cuda80 -c pytorch export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" git clone https://github.com/pytorch/pytorch -b nightly2 --recursive diff --git a/wheel/build_wheel.sh b/wheel/build_wheel.sh index d7f9c13a8..d2bc4d905 100755 --- a/wheel/build_wheel.sh +++ b/wheel/build_wheel.sh @@ -138,7 +138,7 @@ export MACOSX_DEPLOYMENT_TARGET=10.10 export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"} if [[ "$desired_python" == 3.8 ]]; then - retry conda install -yq cmake numpy=1.15 nomkl setuptools pyyaml ninja + retry conda install -yq cmake numpy=1.17 nomkl setuptools pyyaml ninja else retry conda install -yq cmake numpy==1.11.3 nomkl setuptools pyyaml cffi typing ninja requests fi