CUda11.0 安装TENSORRT ubuntu 20.04
时间: 2024-09-09 16:15:49 浏览: 91
CUDA 11.0 是NVIDIA推出的用于GPU计算的并行计算平台和编程模型,而TensorRT是NVIDIA开发的深度学习推理优化器。要在一个Ubuntu 20.04系统上安装CUDA 11.0和TensorRT,可以按照以下步骤操作:
1. **更新系统**:
打开终端,运行以下命令来更新系统包列表并升级所有已安装的包:
```bash
sudo apt update
sudo apt upgrade
```
2. **安装CUDA 11.0**:
由于Ubuntu 20.04的官方仓库中可能没有包含CUDA 11.0,你可能需要下载NVIDIA提供的CUDA 11.0的安装包。访问NVIDIA官方网站下载适合Ubuntu 20.04的CUDA 11.0的.run文件。下载完成后,使用以下命令安装CUDA:
```bash
sudo sh cuda_11.0.3_450.51.06_linux.run
```
安装过程中,你需要同意许可协议,选择安装路径,并根据需要选择组件(比如,是否安装驱动和CUDA Toolkit等)。
3. **安装驱动**:
如果在安装CUDA时没有安装驱动,或者你需要单独安装更新的驱动,你可以访问NVIDIA驱动下载页面,下载适合你的GPU的驱动,并执行.run文件进行安装。例如:
```bash
sudo sh NVIDIA-Linux-x86_64-450.51.run
```
通常推荐安装与CUDA版本相匹配的驱动版本。
4. **安装cuDNN**:
cuDNN是NVIDIA的深度神经网络库,对于TensorRT的安装是必须的。你需要从NVIDIA的官方网站下载cuDNN的Linux版本,解压并安装:
```bash
tar -xzvf cudnn-linux-x86_64-8.0.4.30_cuda11.0-archive.tar.gz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
```
5. **安装TensorRT**:
同样地,你可能需要从NVIDIA官网下载TensorRT 7.x版本的Ubuntu deb包。下载后,使用以下命令安装:
```bash
sudo apt install ./tensorrt_7.x.x.x*_amd64.deb
```
请替换文件名以匹配你下载的TensorRT版本。
6. **验证安装**:
安装完成后,你可以通过运行以下命令来验证CUDA、cuDNN和TensorRT是否正确安装:
```bash
nvcc --version
cat /usr/include/x86_64-linux-gnu/cudnn_v8.h | grep CUDNN_MAJOR -A 2
dpkg -l | grep tensorrt
```
阅读全文