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