Follow instructions on https://pytorch.org
pip install "git+https://github.com/facebookresearch/pytorchvideo.git"
Modified according to AUTOMATIC1111/stable-diffusion-webui#13985 (change import functional_tensor to functional in augmentations.py)
I installed a local copy with added functionality (will be included in this repo).