NVIDIA TX2移植Caffe深度学习框架

本文详细介绍了如何在NVIDIA TX2上移植Caffe深度学习框架,包括依赖顺序、TX2的刷机步骤、CUDNN的安装与测试,以及在移植过程中遇到的CUDA错误701的解决方法。通过修改Makefile.config和Caffe代码,成功适配CUDNNv8,最后验证了Caffe环境的正确性。
摘要由CSDN通过智能技术生成

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
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值