无法加载图像Python扩展

Question

您好，我收到此错误，我尝试了不同的 torch 和 torchvision 版本，但没有任何效果。请问有什么建议吗？

env/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory
warn(f"Failed to load image Python extension: {e}")

我尝试了不同的 torch 和 torchvision 版本

Answer 1

问题原因

Pytorch 和 Torchvision 的版本与以下链接不对应。根据自己的情况，重新加载Torchvision和Pytorch。

解决问题

1. 输入您的 Anaconda 环境和 pip 列表。显示我们所有的图书馆。你可以在这里看到。对我来说，火炬是 1.11.0。根据上面的链接，对应的torchvision应该是0.12.0。
2. pip3 卸载 torchvision。
3. pip3 安装 torchvision==0.12.0。指定安装torchvision的版本。