乐趣区

关于机器学习:MindSpore易点通漫游世界在Jetson-AGX-Orin上安装MindSpore-170-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 环境。两个都不行。怪事。。这可能是一个头孢的未解之谜。(全文完,谢谢浏览)

退出移动版