jetson填坑-单独安装cuda,cudnn,tensorrt任意适用版本

前言

jetson无法单独安装cuda,cudnn,tensorrt的解决方法,比下载SDK manager刷机安装简单好多倍
这个方法是直接下载deb包安装,deb包安装网站
https://repo.download.nvidia.com/jetson/

单独安装cuda

1

sudo apt-get install cuda-toolkit-10-2

2
安装完成之后,将以下复制到 .bashrc文件末尾

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2

3可以手动打开.bashrc文件,也可以用命令直接打开

sudo gedit  ~/.bashrc

单独安装cudnn

简单方法但可控性低:

1首先查看仓库提供的cuDNN有哪一些。terminal输入

sudo apt-cache policy libcudnn8

2如果没有的话可以查看

sudo apt-cache policy libcudnn7

3以此类推

我NX的JetPack 4.5.1安装cuda-10.2之后libcudnn8只有一个版本.则键入

sudo apt-get install libcudnn8

4即可。若有要选择版本的话

sudo apt-get install libcudnn8=(*.*.*.**+cuda*** )

括号中为你之前查看到的版本列表中选择。

简单方法可选择性高:

根据自己的版本在网站https://repo.download.nvidia.com/jetson/
选择需求版本
使用指令:

扫描二维码关注公众号,回复: 16220328 查看本文章
jtop

查看jetson版本:
在这里插入图片描述
查看仓库提供的版本

在这里插入图片描述
之后搜索下面几个文件:(一般在common下面)例如在这里插入图片描述

安装下面顺序安装

dpkg -i libcudnn8_8.0.2.39-1+cuda10.2_arm64.deb
dpkg -i libcudnn8-dev_8.0.2.39-1+cuda10.2_arm64.deb
dpkg -i libcudnn8-doc_8.0.2.39-1+cuda10.2_arm64.deb

单独安装TensorRT

方法1

和安装cudnn相似
进入网站
https://repo.download.nvidia.com/jetson/
选择版本4.6或者你需要的
搜索下面对应tensosrrt版本文件并进行安装
例如我要安装8.0.1版本我会在网页搜索
在这里插入图片描述
把这些文件都下载下来,安装以下顺序安装:

dpkg -i libnvinfer7_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvinfer-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvinfer-plugin7_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvinfer-plugin-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvonnxparsers7_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvonnxparsers-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvparsers7_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvparsers-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvinfer-bin_7.2.0-1+cuda10.2_arm64.deb
dpkg -i libnvinfer-doc_7.2.0-1+cuda10.2_all.deb
dpkg -i libnvinfer-samples_7.2.0-1+cuda10.2_all.deb
dpkg -i tensorrt_7.2.0.14-1+cuda10.2_arm64.deb
dpkg -i python-libnvinfer_7.2.0-1+cuda10.2_arm64.deb
dpkg -i python-libnvinfer-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i python3-libnvinfer_7.2.0-1+cuda10.2_arm64.deb
dpkg -i python3-libnvinfer-dev_7.2.0-1+cuda10.2_arm64.deb
dpkg -i graphsurgeon-tf_7.2.0-1+cuda10.2_arm64.deb
dpkg -i uff-converter-tf_7.2.0-1+cuda10.2_arm64.deb

或者编写bash文件多运行几遍:

#!/bin/bash

for FILE in `find . -name "*.deb"`
do
   sudo apt install ./$FILE
done

方法2

 sudo apt-cache policy tensorrt

选择版本安装,一般没什么选择,但是胜在简单便捷

sudo apt-get install tensorrt=(*.*.*.**+cuda*** )

坑点:

tensorrt版本一般和python3.6绑定,要是有其他方法和其他python版本能绑定的,希望大佬能发给我一份谢谢

引用:

1111
2222

猜你喜欢

转载自blog.csdn.net/weixin_43541510/article/details/130796360