We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8164cca commit 952ea74Copy full SHA for 952ea74
.github/scripts/unittest-windows/install.sh
@@ -29,8 +29,7 @@ else
29
cudatoolkit="pytorch-cuda=${version}"
30
fi
31
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
+conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia pytorch "${cudatoolkit}" pytest pybind11
34
35
torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
36
echo torch.cuda.is_available is $torch_cuda
0 commit comments