安装Tensorrt,pycuda,cuda,cudnn的记录

ubuntu 安装 Tensorrt
一开始需要安装pycuda,但是目前最新的2021.1版本,安装不好。
所以,不安装最新版,运行这个指令应该会安装的pycuda是2020.1:

pip install 'pycuda<2021.1'

注意:后续选择安装包的时候,一定要看自己的cuda和cudnn版本和对应下载的Tensorrt包是不是一样的(至少要满足版本向下兼容)
安装cuda和cudnn的参考博客:http://t.csdn.cn/f6bbX
安装cudnn的时候报错:

test.c:1:10: fatal error: FreeImage.h: 没有那个文件或目录
 #include "FreeImage.h"
          ^~~~~~~~~~~~~

解决方法:

sudo apt-get install libfreeimage3 libfreeimage-dev

Tensorrt的具体安装流程可以先阅读阅读官方那个文档,写得很详细了已经,其他博客和官方文档没啥区别:https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-821/install-guide/index.html#installing-tar
还可以参考博客:http://t.csdn.cn/7zKZ1

可能会出现报错:

ImportError: libnvinfer.so.8: cannot open shared object file: No such file or directory

解决方案看博客:http://t.csdn.cn/Rc4yb

你可能感兴趣的:(ubuntu,深度学习,linux)