ubuntu18.04下载tensorrt
时间: 2023-09-19 14:12:39 浏览: 130
您可以按照以下步骤在 Ubuntu 18.04 上下载 TensorRT:
1. 访问 NVIDIA 的官方网站,注册或登录您的账户:https://developer.nvidia.com/tensorrt
2. 在 TensorRT 页面上,选择您要下载的版本和操作系统。
3. 点击“下载”按钮,下载 TensorRT 安装包。
4. 解压缩下载的安装包,然后运行安装脚本。
5. 在安装过程中,根据提示进行配置,包括选择安装路径、设置环境变量等。
6. 安装完成后,您可以在终端中输入以下命令来测试 TensorRT 是否成功安装:
```
$ cd /usr/src/tensorrt/samples
$ sudo make
$ cd bin
$ ./sample_mnist
```
以上是在 Ubuntu 18.04 上下载 TensorRT 的一般步骤,如果您遇到任何问题,可以参考 NVIDIA 的官方文档或者在社区寻求帮助。
相关问题
ubantu18.04 安装TensorRT
在Ubuntu 18.04上安装TensorRT(NVIDIA Deep Learning SDK的一部分),你需要先确保系统已更新并有 NVIDIA GPU支持。以下是安装步骤:
1. **更新软件包列表**:
```sh
sudo apt-get update
```
2. **安装依赖项** (如果你的显卡不是NVIDIA,可能需要安装CUDA和cuDNN):
```sh
sudo apt-get install -y curl software-properties-common
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-driver nvidia-cuda-toolkit cudnn
```
或者根据你的GPU型号选择合适的驱动。
3. **访问TensorRT的官方发布页面**:
```
cd /tmp
wget https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/GPG_KEY
wget https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/tensorrt-repo_ubuntu18.04.tar.gz
tar xvf tensorrt-repo_ubuntu18.04.tar.gz
```
4. **添加TensorRT的apt源**:
```sh
sudo dpkg -i ./tensorrt-repo-*.deb
```
5. **更新并安装TensorRT**:
```sh
sudo apt-get update
sudo apt-get install libnvinfer7 libnvinfer-dev
```
6. **检查安装**:
```sh
tensorrt --version
```
ubuntu20.04安装tensorRT
根据提供的引用内容,介绍在Ubuntu20.04上安装TensorRT的步骤如下:
1. 根据安装环境下cuda的版本号,在官网中下载对应版本的tar包,例如:TensorRT-7.2.2.3.Ubuntu-18.04.x86_64-gnu.cuda-11.0.cudnn8.0.tar.gz。
2. 解压tar包,可以使用命令:sudo tar -xzvf TensorRT-8.5.1.7.Linux.x86_64-gnu.cuda-11.8.cudnn8.6.tar.gz。
3. 添加环境变量,可以使用命令:export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/rjg/下载/TensorRT-8.5.1.7/lib ('注意/home/rjg/下载'是解压tar包的地址)。
4. 安装TensorRT,可以使用命令:cd TensorRT-8.5.1.7/python,然后使用pip安装tensorrt-8.5.1.7-cp38-none-linux_x86_64.whl。
5. 验证TensorRT安装是否成功,可以使用命令:python -c "import tensorrt;print(tensorrt.__version__)",若正确应输出版本号: 8.5.1.7。
6. 安装uff,可以使用命令:cd ../uff/,然后使用pip安装uff-0.6.9-py2.py3-none-any.whl。
7. 安装graphsurgeon,可以使用命令:cd ../graphsurgeon/,然后使用pip安装graphsurgeon-0.4.6-py2.py3-none-any.whl。
阅读全文