Я пытаюсь установить Intel Optimized Pytorch по-разному - PullRequest
Новая
       23

Я пытаюсь установить Intel Optimized Pytorch по-разному

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

Я впервые использую pyTorch.Я пытаюсь установить его.Во сколько я могу это сделать?Пожалуйста, укажите шаги для этого.

1 Ответ

2 голосов
/ 27 июня 2019

Вы можете установить PyTorch тремя способами.

  1. Использование pip
  2. Использование conda
  3. Из источника

1. Intel Optimized Pytorch Установка

Установите стабильную версию (v 1.0) для Linux через Pip для Python 3.6. 

pip install https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl

pip install torchvision

2. Установка Conda Pytorch 

conda install pytorch-cpu torchvision-cpu -c pytorch

3. PyTorch Установка из источника

Создать новую среду 

conda create -n <env_name> python=3.6

export CMAKE_PREFIX_PATH=/home/user/.conda/envs/<env_name> 

source activate <env_name>

Установить зависимости 

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

conda install -c conda-forge opencv

conda install Pillow

Получить исходный код PyTorch 

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

cd pytorch

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

Установить PyTorch 

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

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

...