Skip to content

Commit 5a6b564

Browse files
committed
Add missing torchcodec dependency to Windows CI installation script.
1 parent 799d288 commit 5a6b564

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

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

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,10 @@ else
3030
wheel="cu$(python -c "print(''.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
3131
fi
3232
printf "Installing PyTorch\n"
33-
pip install --pre torch --index-url https://download.pytorch.org/whl/${UPLOAD_CHANNEL}/${wheel}
33+
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__)"
3437

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

0 commit comments

Comments
 (0)