在Linux上从源代码构建PyTorch时出错

Question

我从源代码构建PyTorch时遇到问题。我使用Anaconda Python 3.6。我使用conda安装了所有依赖项，并发出“ python setup.py install”命令进行构建。它成功构建了所有文件，但随后失败了”（在caffe2 / contrib / cuda-convnet2的第286行）

请帮助！谢谢！

Answer 1

使用python 3.7的安装对我有用。 尝试以下步骤

   conda create -n <env_name> python=3.7

   export CMAKE_PREFIX_PATH=/home/user/.conda/envs/<env_name> #change the path to yours

   source activate <env_name>  

   conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing

   conda install -c conda-forge opencv

   conda install Pillow

   git clone --recursive https://github.com/intel/pytorch

   cd pytorch

   mv caffe2/contrib/cuda-convnet2/ /tmp   # Fix the bug removing the old package out 

   python setup.py install 2>&1 | tee build.out

然后尝试导入pytorch

cd ~
python
import torch

希望这会有所帮助。