Could this be an issue with how the paths are set in https://github.com/pytorch/vision/blob/main/packaging/post_build_script.sh
#!/bin/bash
LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python packaging/wheel/relocate.py
pip install torchvision-extra-decoders
Or we could be dealing with a version mismatch between torch and torchvision