Skip to content

Commit ea437b3

Browse files
authored
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
1 parent 17a7081 commit ea437b3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

.github/scripts/unittest-windows/install.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,7 @@ else
2929
cudatoolkit="pytorch-cuda=${version}"
3030
fi
3131
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
32-
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest
33-
conda install -y -c conda-forge mkl=2020.4
32+
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest pybind11
3433

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

0 commit comments

Comments
 (0)