win10安裝TensorRT

2.1 下載TensorRT

版本使用TensorRT 7.2.3 for Windows，CUDA版本是11.0，

-：NVIDIA TensorRT 7.x Download | NVIDIA Developer

文件大小約500M，

2.2 配置環境變量

將下載好的文件賦值到C盤根目錄，然后將“C:\TensorRT-7.2.3.4\lib”添加環境變量。如下圖：

2.3 安裝必要的whl文件

在shell里面安裝下面三個whl文件。執行命令：

pip install nvidia-pyindex pip install graphsurgeon pip install uff

1

2

3

2.4 復制dll文件到cuda安裝目錄

復制到：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin，如下圖：

到這里就安裝完成了，下面我們測試一下。

2.5 測試

使用VS2017，打開C:\TensorRT-7.2.3.4\samples\sampleMNIST的sample_mnist.sln，如圖：

然后右鍵sample_mnist項目，右鍵選擇屬性。然后再屬性頁選擇“VC++目錄——可執行文件目錄”，打開“可執行文件目錄”，將“C:\TensorRT-7.2.3.4\lib”添加進去，然后點擊“確定”。

選擇“C/C++——附加包含目錄，點進去添加“C:\TensorRT-7.2.3.4\bin”，然后點擊“確定”。

右鍵“sample_mnist”工程，然后選擇“重新生成”。

看到如下信息，則表明生成成功。

生成后的文件如下圖：

然后執行在data-》mnist下面運行“download_pgms.py”文件，下載數據集。

然后執行調試。如下圖

出現下圖則表明成功。

Windows

版權聲明：本文內容由網絡用戶投稿，版權歸原作者所有，本站不擁有其著作權，亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容，請聯系我們jiasou666@gmail.com 處理，核實后本網站將在24小時內刪除侵權內容。