【pytorch操作問題】torchvision安裝成功，但是import torchvision 還是失敗

問題描述

[root@07131396ff2f ~]# python 
Python 3.7.1 (default, Jul  3 2019, 11:22:37) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-36)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torchvision
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/__init__.py", line 1, in <module>
    from torchvision import models
  File "/root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/models/__init__.py", line 11, in <module>
    from . import detection
  File "/root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/models/detection/__init__.py", line 1, in <module>
    from .faster_rcnn import *
  File "/root/.pyenv/versions/3.7.1/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 "/root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/ops/__init__.py", line 1, in <module>
    from .boxes import nms, box_iou
  File "/root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/ops/boxes.py", line 2, in <module>
    from torchvision import _C
ImportError: /root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/_C.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail36_typeMetaDataInstance_preallocated_7E

解決方法

1. 卸載接的torchvision
2. 重新安裝

[root@07131396ff2f ~]# pip uninstall torchvision
Uninstalling torchvision-0.3.0:
  Would remove:
    /root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision-0.3.0.dist-info/*
    /root/.pyenv/versions/3.7.1/lib/python3.7/site-packages/torchvision/*
Proceed (y/n)? y
  Successfully uninstalled torchvision-0.3.0
[root@07131396ff2f ~]# pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html
Looking in links: https://download.pytorch.org/whl/nightly/cu101/torch_nightly.html
Requirement already satisfied: torch in ./.pyenv/versions/3.7.1/lib/python3.7/site-packages (1.4.0)


Collecting torchvision
  Downloading https://download.pytorch.org/whl/nightly/cu101/torchvision-0.5.0.dev20200102-cp37-cp37m-linux_x86_64.whl (4.1MB)
     |████████████████████████████████| 4.1MB 465kB/s 
Requirement already satisfied: numpy in ./.pyenv/versions/3.7.1/lib/python3.7/site-packages (from torchvision) (1.16.4)
Requirement already satisfied: six in ./.pyenv/versions/3.7.1/lib/python3.7/site-packages (from torchvision) (1.12.0)
Requirement already satisfied: pillow>=4.1.1 in ./.pyenv/versions/3.7.1/lib/python3.7/site-packages (from torchvision) (6.1.0)
Installing collected packages: torchvision
Successfully installed torchvision-0.5.0.dev20200102
[root@07131396ff2f ~]# 
[root@07131396ff2f ~]# 
[root@07131396ff2f ~]# python 
Python 3.7.1 (default, Jul  3 2019, 11:22:37) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-36)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> 
>>> import torchvision 
>>>