文章目录
1. 依赖顺序
Caffe <— CUDNN <— CUDA
2. TX2 刷机
2.1 NVIDIA-SDK-MANAGER 下载
官网下载,访问慢。需要注册账号。
2.2 Nvidia TX2 SDK更新
根据install-with-sdkm-jetson的步骤,下载NVIDIA-SDK.
TX2烧录更新过程中,需要解决DNS污染问题,经过ip138查询所需要的ip,直接存入本地,不需要经过DNS解析流程
VirtualBox设置
虚拟机设置–USB设备–启用USB控制器–USB3.0控制器。需要下载Virtualbox对应的exten-pack才能显示出来USB3.0控制器
在Ubuntu18.04中,使用lsusb
命令,USB链接成功后,可以看到二者中的一个。
- Nvidia Corp:recovery 模式
- NVIDIA LInux for Tegra:烧写成功正常启动后,会显示出来。与前者的不同,是后者是既可以传数据,也是被虚拟出来的一块USB网卡。用来烧写TX2 SDK。
如何进入recovery 模式,请参考tx2系统备份与恢复
2.3 CUDNN测试
下载CUDNN例子程序,编译并运行例子程序来确认SDK安装是否成功。
从CUDNN官网https://developer.nvidia.com/rdp/cudnn-download下载最新对应的CUDNN。TX2板子上cuda版本为10.2,所以下载cuDNN8.1.0
下载到CUDNN例子的deb包,对deb包进行解压缩。参考Ubuntu下deb包的解压、打包、安装、卸载及常用命令参数
解压缩后,将源代码例子程序拷贝到TX2开发板上,编译并执行程序。
程序运行正确,说明CUDNN安装正确。并发现该mnistCUDNN是一个训练好的神经网络,而且该神经网络的训练过程实现来自caffe库。
3. caffe库移植
官方下载地址http://caffe.berkeleyvision.org/
3.1 修改Makefile.config
从Makefile.config.example拷贝产生Makefile.config,用来配置caffe编译参数
## Refer to http://caffe.berkeleyvision.org/installation.html
# Contributions simplifying and improving our build system are welcome!
# 安装依赖,参考安装文档进行 依赖 CUDA