anaconda - 在 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

希望这可以帮助。