Failed to load image Python extension: libtorch

1. 问题描述运行一个Pytorch的深度学习工程，当安装好各项以来之后，运行程序，出现如下错误：
/root/anaconda3/envs/yolox/lib/python3.9/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python tension: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory
warn(f"Failed to load image Python extension: {e}")
以上错误提示中，指出了该错误的发生可能是由于pytorch和torchvision班版本不兼容导致。所以，我们来测试一下torchvision:
# 进入python环境python# 加载torchvisionimport torchvision 执行import torchvision时报错：
与上面的报错是一致的，这就印证了之前的假设，所以接下来会从pytorch和torchvision的版本兼容上着手，而不是去想办法安装提示中未找到的动态库libtorch_cuda_cu.so 。
2. torchvision是啥东东？在解决问题之前，先岔开话题扫盲一下torchvision是啥东东，以下是截取的Pytorch中文文档中的解释：
torchvision包包含了目前流行的数据集，模型结构和常用的图片转换工具。

MNIST
COCO（用于图像标注和目标检测）(Captioning and Detection)
LSUN Classification
ImageFolder
Imagenet-12
CIFAR10 and CIFAR100
STL10

torchvision.models模块的子模块中包含以下模型结构。

AlexNet
VGG
ResNet
SqueezeNet
DenseNet

另外，torchvision还包括常用的数据(图片、tensor等)转换工具，详见：torchvision.transforms - PyTorch中文文档
3. 问题解决方法我们到pytorch官网：Start Locally | PyTorch
根据自己的机器环境选择合适的选项：
【Failed to load image Python extension: libtorch】
根据我们的选项，会自动生成安装命令行：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 执行以上安装命令之前，先执行pip3 uninstall torch torchivision将原来已经安装的版本卸载掉。
安装完成后，再次执行测试demo，通过！