我正在尝试以不同的方式安装Intel优化的Pytorch

Question

我是第一次使用pytorch。我正在尝试安装它。我可以通过几种方式做到这一点？请为此提供步骤。

Answer 1

您可以以3种方式安装Pytorch。

1. 使用pip
2. 使用conda
3. 来自来源

1. Intel优化Pytorch安装

在Linux上通过PIP安装稳定版本(v 1.0(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

希望这对您有帮助。