tensorrt安裝windows
以下是在 Windows 上安裝 TensorRT 的詳細步驟:
- 前提條件檢查:
- 擁有英偉達顯卡,且顯存 8GB 以上(30 系、40 系等新卡更好)。
- 安裝了英偉達顯卡驅動,版本需大于等于 530。
- 安裝 CUDA:
- 根據 TensorRT 的版本要求,下載兼容的 CUDA 版本。例如,可前往英偉達官方網站的 CUDA 下載頁面,選擇適合您系統的版本(如 CUDA 12.1 等)。下載完成后,雙擊安裝程序,安裝位置可選擇默認。
- 安裝完成后,確認 CUDA 的庫文件路徑和二進制文件路徑已添加到系統的環境變量中。您可以在命令提示符中輸入
nvcc -V來檢查 CUDA 是否安裝成功。
- 安裝 cuDNN:
- 在英偉達官方網站上下載與您安裝的 CUDA 版本相匹配的 cuDNN。比如,如果您安裝的是 CUDA 12.1,就選擇適合 CUDA 12.x 的 cuDNN 版本。
- 將下載的 cuDNN 壓縮包解壓,找到 CUDA 的安裝目錄(例如
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1),然后將 cuDNN 解壓后的對應文件夾中的文件復制到 CUDA 安裝目錄下的相應文件夾。具體來說,需要復制 3 次,分別將 cuDNN 的bin文件夾下的文件復制到 CUDA 的bin文件夾,include文件夾下的文件復制到 CUDA 的include文件夾,lib文件夾下的文件復制到 CUDA 的lib文件夾。
- 安裝 TensorRT:
- 從英偉達官方網站下載 TensorRT 的壓縮包。需注意選擇與您的 CUDA 版本和系統相匹配的版本。
- 解壓下載的 TensorRT 壓縮包。
- 將 TensorRT 解壓后的
bin文件夾中的所有動態鏈接庫(.dll 文件)復制到 CUDA 的bin文件夾。 - 將 TensorRT 的
include文件夾復制到 CUDA 的include文件夾。 - 將 TensorRT 的
lib文件夾中的靜態庫(.lib 文件)復制到 CUDA 的lib文件夾。
- 安裝 TensorRT 的 Python 包(如果需要在 Python 中使用):
- 進入解壓后的 TensorRT 文件夾中的
python子文件夾。 - 使用
pip命令安裝該文件夾下對應的.whl文件。例如,如果您的 Python 版本是 3.8,就安裝tensorrt-xxx-cp38-none-win_amd64.whl(其中xxx為具體的版本號)。您可能還需要安裝graphsurgeon、uff、onnx_graphsurgeon等相關的 Python 包。
- 進入解壓后的 TensorRT 文件夾中的
- 驗證安裝:
- 打開命令提示符或 Python 解釋器,輸入
import tensorrt,如果沒有報錯并能正確打印出 TensorRT 的版本信息,則說明安裝成功。
- 打開命令提示符或 Python 解釋器,輸入

