我在从源代码构建 PyTorch 时遇到问题。我使用 Anaconda Python 3.6。我使用 conda 安装了所有依赖项并发出“python setup.py install”命令来构建它。它成功构建了所有文件,但随后失败“在 caffe2/contrib/cuda-convnet2 的第 286 行”
请帮忙!谢谢!
我在从源代码构建 PyTorch 时遇到问题。我使用 Anaconda Python 3.6。我使用 conda 安装了所有依赖项并发出“python setup.py install”命令来构建它。它成功构建了所有文件,但随后失败“在 caffe2/contrib/cuda-convnet2 的第 286 行”
请帮忙!谢谢!
使用 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
希望这可以帮助。