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的版本兼容上着手，而不是去想办法安装提示中未找到的动态库l