ubunutu20.04安装tensorrt
时间: 2024-06-15 16:04:12 浏览: 241
TensorRT安装
Ubuntu 20.04是一种流行的Linux操作系统,而TensorRT是英伟达(NVIDIA)推出的用于高性能深度学习推理的库。下面是在Ubuntu 20.04上安装TensorRT的步骤:
1. 首先,确保你的系统已经安装了NVIDIA显卡驱动。你可以通过运行以下命令来检查驱动程序是否已正确安装:
```
nvidia-smi
```
2. 接下来,下载并安装TensorRT。你可以从NVIDIA的官方网站上下载适用于Ubuntu 20.04的TensorRT安装包。下载完成后,使用以下命令进行安装:
```
sudo dpkg -i <path_to_tensorrt_package.deb>
sudo apt-get update
sudo apt-get install tensorrt
```
3. 安装完成后,你可以验证TensorRT是否成功安装。运行以下命令来查看TensorRT的版本信息:
```
dpkg -l | grep TensorRT
```
4. 接下来,你需要设置TensorRT的环境变量。打开终端并编辑`~/.bashrc`文件:
```
nano ~/.bashrc
```
在文件末尾添加以下行:
```
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/x86_64-linux-gnu/
export PATH=$PATH:/usr/lib/x86_64-linux-gnu/
```
保存并关闭文件,然后运行以下命令使更改生效:
```
source ~/.bashrc
```
至此,你已经成功在Ubuntu 20.04上安装了TensorRT。
阅读全文