电脑环境是
toch1.2
torchvision0.3.0
cuda10.0
在运行
import torchvision
会报错:
libcudart.so.9.0: cannot open shared object file: No such file or directory
解决方法一
降低torchvision的版本为torchvision0.2.2,错误解除,
可是我需要用到torchvision0.3.0里面的object detection包,
所以,该方法并没有解决本质问题。
解决方法二
本质原因:torchvision0.3支持CUDA9,不支持10,更新至torchvison0.4即可;
更新torchvision到0.4会带着torch版本自动更新,
那么可以先下载torchvision0.4的安装包,
然后终端执行:
pip install torchvision-0.4.0+cu92-cp36-cp36m-manylinux1_x86_64.whl
即可。
torchvision-0.4.0+cu92-cp36-cp36m-manylinux1_x86_64.whl 为我下载到本地的torchvision0.4安装包
错误解除!