下載TensorRT
官網: https://developer.nvidia.com/tensorrt
下載位置: https://developer.nvidia.com/tensorrt/download/10x (需要登入)
解壓縮
解壓縮後將TensorRT資料夾放合適的位置
新增環境變數
path: D:\tools\TensorRT-10.6.0.26\lib
D:\tools\TensorRT-10.6.0.26\bin
將lib中的dll複製到cuda的bin目錄下
測試
pip install tensorrt
import tensorrt
print(tensorrt.__version__)
參考文件: https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html