ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory

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