当前环境: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