torchvision版本问题。
File "/home/qq/anaconda3/envs/qq/lib/python3.7/site-packages/torchvision/models/detection/faster_rcnn.py", line 7, in <module>
from torchvision.ops import misc as misc_nn_ops
File "/home/qq/anaconda3/envs/qq/lib/python3.7/site-packages/torchvision/ops/__init__.py", line 1, in <module>
from .boxes import nms, box_iou
File "/home/qq/anaconda3/envs/qq/lib/python3.7/site-packages/torchvision/ops/boxes.py", line 2, in <module>
from torchvision import _C
ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory
更新到0.3.0之后,就开始报错。
很奇怪的是,我的cuda是10.0的,这里却报的9.0,不懂。
但是pip install torchvision == 0.2.2之后就能正常运行了。
ubuntu 18.04.2
python 3.7.1
torch 1.1.0
cuda 10.0