現在caffe2被合並到了PyTorch中
git clone https://github.com/pytorch/pytorch
pip install -r requirements.txt
sudo python setup.py install
后邊報錯信息的解決
遇到
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "torch/__init__.py", line 84, in <module>
from torch._C import *
ImportError: No module named _C
錯誤的時候,有兩種可能:
numpy版本不匹配,使用sudo pip install numpy -I重新安裝numpy
找不到_C.so文件。到/usr/local/lib/python2.7/dist-packages下查看是否存在_C.so(文件名必須是這個,如果文件名太長,需要修改成這個)
之后切換到自己目錄下重新嘗試后,問題解決