TensorRT系列(一)環境設定

本文章版本資訊

  • TensorRT 8.6.1

apt 安裝TensorRT的方法

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

利用Docker 建立TensorRT環境

docker tensorrt的版本與tensorrt版本對應表
https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html

1
docker run --gpus all -it --rm -v local_dir:container_dir nvcr.io/nvidia/tensorrt:22.07-py3

其中local_dir換成自己電腦的某一個資料夾,他將與Docker container共用,container_dir換成container裡面的資料夾路徑

參考:
https://docs.nvidia.com/deeplearning/tensorrt/quick-start-guide/index.html#install
https://github.com/wang-xinyu/tensorrtx#tutorials

Author

Steven

Posted on

2022-08-24

Updated on

2025-01-14

Licensed under

Comments