【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环境。两个都不行。怪事。。这可能是一个头孢的未解之谜。(全文完,谢谢浏览)