如何在Power 8或PPC64机器上安装pytorch?

时间:2018-10-11 00:49:29

标签: pytorch powerpc

我正在尝试使用 conda power 8 IBM机器上安装 pytorch 。虽然,我从IBM blog阅读了一些文章,但由于无法坚持编译岩浆,所以无法成功安装。

2 个答案:

答案 0 :(得分:2)

假定已安装 conda 。 只需运行以下命令

conda install -c engility pytorch
conda install -c engility torchvision

注意:
1.转到此anaconda
2.搜索火炬
3.向下滚动以查看哪个具有 linux-ppc64le 作为平台
4.单击进入该特定程序包
5.您将获得安装pytorch的命令

答案 1 :(得分:1)

我最终使用conda环境在ppc64le服务器上安装了PyTorch 1.5.0。

  1. 安装 anaconda (在我的情况下,我刚刚安装了miniconda3):
  2. 添加新配置频道
conda config --prepend channels https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda-early-access/linux-ppc64le/
  1. 创建一个新的conda 环境并激活它:
conda create --name early_access python=3.6
conda activate early_access
  1. 安装火炬
conda install pytorch

就是这样!归功于此博客:https://gauravm.gitbook.io/about/blogs/installing-pytorch-and-transformers-on-ibm-powerpc-architecture