TensorRT7安装
前言
但是本机原来安装的是CUDA10.1,但是TensorRT7的最新版本需要CUDA10.2,所以先重新安装一个CUDA。
TIPs: 因为之前在win平台上使用过tensorrt,得到的唯一一个结论就是:千万要对应版本!!!无论是安装还是后续的部署,tensorrt都需要cuda和平台类型保持一致。
安装CUDA10.2
检查系统信息
- 包括系统版本、GPU型号,确认安装合适版本的CUDA
- 检查gcc版本
gcc --version
# 或者
gcc -v
之前有看到一个说法,CUDA10.2不支持gcc 9,所以最好检查一下。
这里有个问题,我通过ssh连接服务器,但是ssh的终端和直接在服务器上打开的终端显示的内容不一样。
在ssh上能够用gcc --version查询gcc版本,但是直接在服务器终端上会报错,显示gcc找不到'--version'这个程序。
但是使用gcc -v这个指令都能查到gcc版本。
下载CUDA安装包
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
然

本文介绍了在Ubuntu18.04系统上,由于TensorRT7需要CUDA10.2,因此需要卸载原有CUDA10.1并安装CUDA10.2。详细步骤包括检查系统信息、下载CUDA安装包并安装、设置环境变量,接着安装CuDNN并验证,最后下载TensorRT7,解压并配置环境变量,通过sampleMNIST测试安装是否成功。
最低0.47元/天 解锁文章
447

被折叠的 条评论
为什么被折叠?



