第一次接触踩了不少坑,切忌不要按照常见服务器、电脑的思路安装。
安装 JetPack 套件
JetPack 是 Nvidia为 Jetson 系列开发板开发的一款软件开发包,常用的开发工具基本都有,安装 Jetson 会自动的将匹配版本的CUDA、cuDNN、TensorRT等安装好,极大地方便了开发者!几乎就是一键傻瓜安装,爽到爆!
sudo apt update
sudo apt install nvidia-jetpack
sudo apt show nvidia-jetpack
安装pytorch with cuda
安装完 JetPack 后会自动安装cuda,pytorch不能直接装pytorch官网的,需要修改与 JetPack 对应的版本,更多版本详见pytorch。下面圈红的是我安装的版本,我用的是 JetPack 5.1,cuda版本对应的是11.4,cuda版本可以通过nvcc -V查看。
点击箭头的位置开始下载.whl文件夹,下载后在.whl文件夹文件对应的文件夹内输入下面命令即可安装