Warm tip: This article is reproduced from serverfault.com, please click

nvidia-如何完成tensorRT zip安装过程的第三步?

(nvidia - how to complete the 3rd step of tensorRT zip installation process?)

发布于 2020-12-15 15:17:57

我无法理解整个过程,因为它说我必须替换特定下载的tensorRT,cuda和cuDNN版本。他们正在谈论什么特别的下载?

https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-zip

这是第3步的完整步骤。请通过解释进行说明:

将TensorRT-7.xxxWindows10.x86_64.cuda-xxcudnnx.x.zip文件解压缩到你选择的位置。代替:

7.x.x.x with the TensorRT version
cuda-x.x with the CUDA version, and
cudnnx.x with the cuDNN version for your particular download.

PS:TensorFlow和tensorRT机器学习的新手。

Questioner
Ghost
Viewed
11
Nyps 2020-12-20 05:00:30

你实际上不需要“替换”任何东西,而是x字母。你将通过选择适合你的系统的TensorRT版本来自动执行此操作。

例子:

在安装TensorRT之前,你必须安装CUDA和cuDNN的版本。

对于Windows 10,根据此处,应为:

CUDA 10.2, 11.0 update 1, 11.1 update 1, or 11.2
cuDNN 8.0.5

假设你没有安装CUDA 10.2cuDNN 8.0.5并要安装TensorRT 7.2.1.6然后,你要查找的文件将被命名为:TensorRT-7.2.1.6.Windows10.x86_64.cuda-10.2.cudnn8.0.zip

这是你应该下载并解压缩的文件。