Получение ошибки при сборке PyTorch из исходного кода в Linux - PullRequest
Новая
       18

Получение ошибки при сборке PyTorch из исходного кода в Linux

0 голосов
/ 28 июня 2019

У меня проблема со сборкой PyTorch из исходного кода.Я использую Anaconda Python 3.6.Я установил все зависимости, используя conda, и выполнил команду «python setup.py install» для его сборки.Он успешно строит все файлы, но затем терпит неудачу "в строке 286 файла caffe2 / contrib / cuda-convnet2"

Pls help!Спасибо!

1 Ответ

0 голосов
/ 09 июля 2019

У меня установка с 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

Надеюсь, это поможет.

...