英伟达Jetson AGX Orin 64G使用初探
概要
笔者拿到英伟达的Jetson AGX Orin时已经预装了配套的ubuntu系统,其实是内置了CUDA的开发环境的,只需配置一下环境就可以用了,因为我只需要使用cuda开发算子,没有用到深度学习这些模型框架,也就没有安装,后面有需要可以再安装呗。
步骤1 配置CUDA环境
在桌面打开终端,输入以下命令:
vim ~/.bashrc
打开后,在文本里面添加如下路径:
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_HOME=/usr/local/cuda
然后esc,wq保存退出,再更新环境变量配置:
source ~/.bashrc
接着输入nvcc -V查看版本号。
nvcc -V
返回CUDA版本号即大功告成了。
安装jtop
设备联网,安装JTOP工具,用于在开发过程中监控硬件。
sudo -H pip install jetson-stats
安装成功后,输入jtop,出现一下界面即成功了
接下里就快乐的开启你的CUDA开发之旅吧,恭喜入坑。
小结
以后我将就我在项目里的开发经验,分享一些GPU开发和优化的技巧以及一些踩坑指南,还有国产GPU的一些使用心得;从安装环境,到配置开发环境、再到编译套件的编写、到算子的开发调优等,既是分享,也是个人的复习