我正在尝试以不同的方式安装英特尔优化的 pytorch

I am trying to install intel optimized pytorch in different ways

我是第一次使用pyTorch。我正在尝试安装它。我可以通过多少种方式做到这一点? 请提供相关步骤。

您可以通过 3 种方式安装 PyTorch。

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

1.Intel优化Pytorch安装

通过 Pip 为 Python 3.6 在 Linux 上安装稳定版本 (v 1.0)。

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

希望对您有所帮助。