Failed to load image Python extension: libtorch_cuda_cu.so

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

# 加载torchvision
import 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

 根据自己的机器环境选择合适的选项：

 根据我们的选项，会自动生成安装命令行：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

执行以上安装命令之前，先执行pip3 uninstall torch torchvision将原来已经安装的版本卸载掉。

安装完成后，再次执行测试demo，通过！