c++ 环境tensorrt安装
时间: 2023-05-23 17:05:55 浏览: 90
A:
首先,确保您的系统中已经安装了CUDA和cuDNN。然后,按照以下步骤安装 TensorRT:
1. 下载TensorRT压缩包并解压:
```
$ wget https://developer.nvidia.com/compute/machine-learning/tensorrt/7.0/ga/local_repos/nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb
$ sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb
$ sudo apt-key add /var/nv-tensorrt-repo-cuda10.2-trt7.0.0.11-ga-20191216/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install tensorrt
```
2. 下载并解压 TensorRT 的样例:
```
$ mkdir -p $HOME/tensorrt/samples
$ cd $HOME/tensorrt/samples
$ cp -r /usr/src/tensorrt/samples/* .
```
3. 编译样例:
```
$ cd $HOME/tensorrt/samples
$ make
```
4. 运行样例:
```
$ cd $HOME/tensorrt/samples/bin
$ ./sample_mnist
```
如果运行正常,您应该能够看到输出类似于:
```
Test Case: test
Accuracy: 0.990500
```