龙芯loongarch64服务器编译安装pytorch

前言

        PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序,它是一个基于Python的可续计算包。在通过龙芯开源仓库下载依赖包后,执行import torch后,发现其中安装的"pytorch"就无法正常使用,这里就演示下pytorch整个编译流程。

龙芯loongarch64服务器编译安装pytorch_第1张图片

1、环境准备

如果要从源代码安装,则需要:

  • Python 3.8 或更高版本(对于 Linux,需要 Python 3.8.1+)
  • 完全支持 C++17 的编译器,例如 clang 或 gcc(特别是对于 aarch64、gcc 9.4.0 或更高版本是必需的),实际测试系统本身自动gcc的就可以正常编译。

Python版本默认安装的是python3.7,可以看我之前的文章龙芯loongarch64服务器编译安装3.8版本的Python

系统相关依赖包下载安装:

yum install ninja*

yum install cmake

yum install gcc*

2、源码拉取

目前使用Pytorch 2.1.0版本的,低版本的暂不考虑。

git clone -b v2.1.0 https://github.com/pytorch/py

你可能感兴趣的:(龙芯loongarch64,pytorch,人工智能,python)