PyTorch安装问题解决

Posted 2021-01-09 jourluohua

现在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(文件名必须是这个，如果文件名太长，需要修改成这个)

之后切换到自己目录下重新尝试后，问题解决