Unverified Commit ea437b31 authored by ahmadsharif1's avatar ahmadsharif1 Committed by GitHub
Browse files

Fix windows CI CPU and GPU tests by unpinning mkl version and installing pybind11 (#3766)

Unpin mkl version and install pybind11 to get the windows CI working again

This fixes #3767
parent 17a70815
...@@ -29,8 +29,7 @@ else ...@@ -29,8 +29,7 @@ else
cudatoolkit="pytorch-cuda=${version}" cudatoolkit="pytorch-cuda=${version}"
fi fi
printf "Installing PyTorch with %s\n" "${cudatoolkit}" printf "Installing PyTorch with %s\n" "${cudatoolkit}"
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest pybind11
conda install -y -c conda-forge mkl=2020.4
torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())") torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
echo torch.cuda.is_available is $torch_cuda echo torch.cuda.is_available is $torch_cuda
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment