在NVIDIA Jetson计算平台中创建Tensorrt容器

当前环境:jetpack4.4DP、jetson nano

注:适用于jetpack>=4.2和其他jetson开发板,如Xavier AGX、NX、TX2


前言

需要在jetpack4.4DP环境的Jetson开发板里安装Tensorrt的docker容器(Tensorrt基于jetpack4.4DP)。官网提供的是Dockerfile的安装方式,详情见官网。不知道NVIDIA有没有更新这个bug:拉取的基础镜像是基于X86架构,而Jetson是基于ARM架构。安装难点在于Tensorrt官网上没有提供基于Jetson的单独安装包,所以对于这一块就比较陌生。下面是我的解决方案,仅供参考。此方案可适用于jetpack的其他版本,只需要在配置环境时选择自己要用的版本即可。

一、拉取docker基础镜像卸载基础镜像中的CUDA

1、这里将官方提供的tensorflow镜像作为基础镜像,镜像地址 https://ngc.nvidia.com/catalog/containers/nvidia:l4t-tensorflow,在jetson nano命令窗口输入指令

sudo docker pull nvcr.io/nvidia/l4t-tensorflow:r32.4.3-tf1.15-py3

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值