【MindSpore易点通·漫游世界】在Jetson AGX Orin上安装MindSpore 1.7.0 GPU版

GPU加速云服务器

MindSpore

发表于 2022-07-04 22:59:26304查看

张小白曾经在 Jetson AGX Orin上使用gcc 7.3.0成功源码编译安装过 MindSpore 1.6.0 CUDA11的版本：https://bbs.huaweicloud.com/forum/thread-186292-1-1.html

也刚刚试图使用 gcc 9.4.0源码编译 MindSpore 1.8.0 CUDA11的版本失败：https://bbs.huaweicloud.com/forum/thread-193443-1-1.html

所以不得不退而求其次，安装官方明着显示的最高版本 1.7.0

具体方法如下：

先将 gcc和g++的软连接都指向 /usr/bin/gcc-9和 /usr/bin/g++-9

然后下载mindspore 1.7的源码包：

接着修改build_mindspore.sh

如上图所示，箭头处增加 -DENABLE_GITEE=ON

执行 ./build.sh -e gpu -j12 开始编译。

。。。

编译得非常顺利，让张小白有点觉得不可思议。

张小白几乎忘记了现在到底是用什么Python版本编译的：

原来是Python 3.9

看看编译的结果：

创建一个新的conda环境：

查看下版本：

于是开始安装：

pip install ./mindspore_gpu-1.7.0-cp39-cp39-linux_aarch64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

验证：vi test.py

import numpy as np
from mindspore import Tensor
import mindspore.ops as ops
import mindspore.context as context

context.set_context(device_target="GPU")
x = Tensor(np.ones([1,3,3,4]).astype(np.float32))
y = Tensor(np.ones([1,3,3,4]).astype(np.float32))
print(ops.add(x, y))

python test.py