From 952ea7457bcc3ed0669e7741ff23015c426d6322 Mon Sep 17 00:00:00 2001 From: ahmadsharif1 Date: Mon, 15 Apr 2024 12:44:31 -0400 Subject: [PATCH] [RELEASE-ONLY CHANGES for 2.3] cherry-pick ea437b3 to fix windows CI (#3769) Unpin mkl version and install pybind11 to get the windows CI working again This fixes #3767 --- .github/scripts/unittest-windows/install.sh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/scripts/unittest-windows/install.sh b/.github/scripts/unittest-windows/install.sh index 676de7e830..0dd4997323 100644 --- a/.github/scripts/unittest-windows/install.sh +++ b/.github/scripts/unittest-windows/install.sh @@ -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