iT邦幫忙

1

[Yolov8-TensorRT] - 3. TensorRT安裝

  • 分享至 

  • xImage
  •  
  1. 下載TensorRT
    官網: https://developer.nvidia.com/tensorrt
    下載位置: https://developer.nvidia.com/tensorrt/download/10x (需要登入)
    https://ithelp.ithome.com.tw/upload/images/20241214/20134736y4G9PkNCOq.jpg

  2. 解壓縮
    解壓縮後將TensorRT資料夾放合適的位置
    https://ithelp.ithome.com.tw/upload/images/20241214/20134736219iVXKQvG.jpg

  3. 新增環境變數

path: D:\tools\TensorRT-10.6.0.26\lib
	  D:\tools\TensorRT-10.6.0.26\bin
  1. 將lib中的dll複製到cuda的bin目錄下
    https://ithelp.ithome.com.tw/upload/images/20241214/20134736lONV6Chs4B.jpg
    https://ithelp.ithome.com.tw/upload/images/20241214/20134736Jf2HKP62yJ.jpg

  2. 測試

pip install tensorrt
import tensorrt

print(tensorrt.__version__)

https://ithelp.ithome.com.tw/upload/images/20241214/201347369kwJQCLrIJ.jpg

參考文件: https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html


圖片
  熱門推薦
圖片
{{ item.channelVendor }} | {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言