When installing torchvision with the following command on linux aarch64 with python3.10:
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
The image.so library is missing in the dist-packages folder:
/usr/local/lib/python3.10/dist-packages/torchvision/
Resulting in the following warning when importing torchvision:
torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")
When building torchvision from source the image.so file is at the right location and the import warning is gone. So probably the pip wheel is not created correctly.