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

  • 相关阅读:
    次小生成树(SST)
    传纸条(scrip)
    动态规划练习5
    动态规划练习4
    整数的lqp拆分
    [HNOI2002]跳蚤
    BZOJ1803: Spoj1487 Query on a tree III
    51nod-1526: 分配笔名
    51nod-1615: 跳跃的杰克
    BZOJ2588: Spoj 10628. Count on a tree
  • 原文地址:https://www.cnblogs.com/tibetanmastiff/p/13496891.html
Copyright © 2011-2022 走看看