ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory
错误类型
Traceback ( most recent call last) :
File "/home/ices/lixian/CD/CloudComp/NTS-Net-master/train.py" , line 7 , in < module>
from core import model, dataset
File "/home/ices/lixian/CD/CloudComp/NTS-Net-master/core/dataset.py" , line 7 , in < module>
from torchvision import transforms
File "/home/ices/lixian/CD/torchvision/__init__.py" , line 1 , in < module>
from torchvision import models
File "/home/ices/lixian/CD/torchvision/models/__init__.py" , line 11 , in < module>
from . import detection
File "/home/ices/lixian/CD/torchvision/models/detection/__init__.py" , line 1 , in < module>
from . faster_rcnn import *
File "/home/ices/lixian/CD/torchvision/models/detection/faster_rcnn.py" , line 7 , in < module>
from torchvision. ops import misc as misc_nn_ops
File "/home/ices/lixian/CD/torchvision/ops/__init__.py" , line 1 , in < module>
from . boxes import nms, box_iou
File "/home/ices/lixian/CD/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
解决方案
torchvision 版本不对 将torchvision版本改为0.2.2
pip install torchvision== 0.2 .2