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

最新推荐文章于 2024-09-24 14:51:03 发布

黄沛鑫HPX

最新推荐文章于 2024-09-24 14:51:03 发布

阅读量824

点赞数 2

文章标签： ubuntu 深度学习 linux

本文链接：https://blog.csdn.net/jerry18863/article/details/127301210

版权

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

关注