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

zi8p0yeb  于 2022-12-13  发布在  其他
关注(0)|答案(3)|浏览(190)

我尝试使用condapower 8 IBM机器上安装pytorch。虽然我阅读了IBM blog的文章,但我无法成功安装,因为我遇到了编译magma的问题。

ctehm74n

ctehm74n1#

假设已经安装了conda

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

注意事项:
1.后藤此anaconda
2.搜索pytorch
3.向下滚动以查看哪一个将linux-ppc 64 le作为平台
4.单击该特定包
5.您将获得安装pytorch的命令

wko9yo5t

wko9yo5t2#

我最终使用conda环境在ppc 64 le服务器上安装了PyTorch 1.5.0。
1.安装anaconda(在我的例子中,我只安装了miniconda 3):
1.添加新配置通道

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.安装pytorch

conda install pytorch

就是这样!这个博客的作者:https://gauravm.gitbook.io/about/blogs/installing-pytorch-and-transformers-on-ibm-powerpc-architecture

ecbunoof

ecbunoof3#

现在您可以使用powerai - MLDL framework from IBM,它包含PyTorch:

conda install powerai

它会安装PPC64的正确版本pytorch。

相关问题