Skip to content

Commit

Permalink
[RELEASE-ONLY CHANGES for 2.3] cherry-pick ea437b3 to fix windows CI (#…
Browse files Browse the repository at this point in the history
…3769)

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

This fixes #3767
  • Loading branch information
ahmadsharif1 authored Apr 15, 2024
1 parent 8164cca commit 952ea74
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions .github/scripts/unittest-windows/install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,7 @@ else
cudatoolkit="pytorch-cuda=${version}"
fi
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest
conda install -y -c conda-forge mkl=2020.4
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest pybind11

torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
echo torch.cuda.is_available is $torch_cuda
Expand Down

0 comments on commit 952ea74

Please sign in to comment.