We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 799d288 commit 5a6b564Copy full SHA for 5a6b564
.github/scripts/unittest-windows/install.sh
@@ -30,7 +30,10 @@ else
30
wheel="cu$(python -c "print(''.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
31
fi
32
printf "Installing PyTorch\n"
33
-pip install --pre torch --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${wheel}
+pip install --pre torch torchcodec --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${wheel}
34
+
35
+python -c "import torch; print(torch.__version__)"
36
+python -c "import torchcodec; print(torchcodec.__version__)"
37
38
torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
39
echo torch.cuda.is_available is $torch_cuda
0 commit comments