龙芯3A6000编译PyTorch 2.1.0（含Python 3.8和CMake 3.27.7）

最新推荐文章于 2024-04-06 10:18:01 发布

SmileyFaceApp

最新推荐文章于 2024-04-06 10:18:01 发布

阅读量728

点赞数 2

文章标签： python pytorch

本文链接：https://blog.csdn.net/zhangxiangxiao/article/details/134193265

版权

本文详细介绍了如何在龙芯3A6000 Loongnix 20系统上，从编译Python 3.8、CMake 3.27.7到成功编译PyTorch 2.1.0的过程，以适配LoongArch64架构。文章提供了具体的步骤和注意事项，包括修改Python源代码以识别新架构，以及利用龙芯开源社区的Python软件源解决依赖问题。

摘要由CSDN通过智能技术生成

近日收到了龙芯3A6000的产品机器，配了8GB的内存和4GB Radeon独显。系统为桌面版Loongnix 20，是龙芯开源社区推出的Linux发行版。不同于之前的龙芯CPU，龙芯3A6000采用了全新的LoongArch64架构，因此软件生态还不甚完善。本文将在Loongnix 20环境中编译PyTorch 2.1.0，以使其能运行于龙芯3A6000 CPU上。

本文假设系统用户名为龙芯系统默认的sancog，路径以用户文件夹/home/sancog为基础。

1. 编译Python 3.8

因Loongnix是基于Debian稳定版发布的，其自含的Python版本较旧，为Python 3.7。由于PyTorch 2.1.0需要的Python版本最小为3.8，因此我们需要首先编译Python 3.8。我们首先下载安装编译需要的环境。

sancog@sancog-pc:~$ sudo apt install build-essential wget

然后下载并解压Python 3.8的官方源代码包。

sancog@sancog-pc:~$ wget https://www.python.org/ftp/python/3.8.18/Python-3.8.18.tar.xz
sancog@sancog-pc:~$ tar -xvf Python-3.8.18.tar.xz

由于Python 3.8代码包的编译设置环境不识别LoongArch64架构，我们需要对其进行修改。首先我们替换最新的可以识别LoongArch*架构的config.sub和config.guess文件。

sancog@sancog-pc:~$ cd Python-3.8.18
sancog@sancog-pc:~/Python-3.8.18$ wget -O config.sub "git.savannah.gnu.org/gitweb/?p=config.git;a=blob_plain;f=config.sub;hb=HEAD"
sancog@sancog-pc:~/Python-3.8.18$ wget -O config.guess "git.savannah.gnu.org/gitweb/?p=config.git;a=blob_plain;f=config.guess;hb=HEAD"

其次，我们修改configure文件使其能够生成正确的PLATFORM_TRIPLET及EXT_SUFFIX变量