zoukankan      html  css  js  c++  java
  • pytorchvision安装问题

    问题1:找不到cuda库

    Traceback (most recent call last):
      File "cnn.py", line 5, in <module>
        import torchvision
      File "/workspace/jfyuan/tools/espnet/tools/venv/lib/python3.7/site-packages/torchvision/__init__.py", line 1, in <module>
        from torchvision import models
      File "/workspace/jfyuan/tools/espnet/tools/venv/lib/python3.7/site-packages/torchvision/models/__init__.py", line 11, in <module>
        from . import detection
      File "/workspace/jfyuan/tools/espnet/tools/venv/lib/python3.7/site-packages/torchvision/models/detection/__init__.py", line 1, in <module>
        from .faster_rcnn import *
      File "/workspace/jfyuan/tools/espnet/tools/venv/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 "/workspace/jfyuan/tools/espnet/tools/venv/lib/python3.7/site-packages/torchvision/ops/__init__.py", line 1, in <module>
        from .boxes import nms, box_iou
      File "/workspace/jfyuan/tools/espnet/tools/venv/lib/python3.7/site-packages/torchvision/ops/boxes.py", line 2, in <module>
        from torchvision import _C
    ImportError: /workspace/jfyuan/pytorch/sample/cnn/libs/libcudart.so.9.0: version `libcudart.so.9.0' not found (required by
    

    原因:pytorch是1.3.1,pytorchvision安装的0.3.0,版本不对应导致的

    解决:pip install torchvision==0.4.2

  • 相关阅读:
    redis集群
    鉴权方案选择
    spring mvc 自定义handler不拦截静态资源
    servlet3
    压测工具 ab jmeter
    死锁产生的原因
    缓存方案:本地guavaCache, 远程redis?
    使用spring boot admin
    groovy使用小记
    python--面试题01
  • 原文地址:https://www.cnblogs.com/tibetanmastiff/p/13496891.html
Copyright © 2011-2022 走看看