Skip to content

Commit

Permalink
Fix windows CI CPU and GPU tests by unpinning mkl version and install…
Browse files Browse the repository at this point in the history
…ing pybind11 (#3766)

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

This fixes #3767
  • Loading branch information
ahmadsharif1 committed Mar 25, 2024
1 parent 8164cca commit 6c6e336
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 6c6e336

Please sign in to comment.