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

  • 相关阅读:
    在intent-filter中的配置
    利用asynchttpclient开源项目来把数据提交给服务器
    URL的应用
    ScrollView在布局中的作用
    android中传统的创建数据库
    for (Sms sms : smsLists){}
    Android中对文件的读写进行操作
    android中的5大布局
    android复制包需要修改的几个地方
    【Unity Tips】备忘录(扫盲篇)
  • 原文地址:https://www.cnblogs.com/tibetanmastiff/p/13496891.html
Copyright © 2011-2022 走看看