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

【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/f...也刚刚试图使用 gcc 9.4.0源码编译 MindSpore 1.8.0 CUDA11的版本失败：https://bbs.huaweicloud.com/f...所以不得不退而求其次，安装官方明着显示的最高版本 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.ed...

验证：vi test.pyimport 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

奇怪了，退回到base看看：conda deactivate python test.py

python3 test.py

貌似是python和python3有多个版本的问题。

貌似在base环境，怎么都可以。但是在conda的orin python 3.9环境。两个都不行。怪事。。这可能是一个头孢的未解之谜。（全文完，谢谢阅读）