Windows TensorRT环境配配置（VS Code）

最新推荐文章于 2024-08-12 23:05:20 发布

蘑菇爸爸

最新推荐文章于 2024-08-12 23:05:20 发布

阅读量799

点赞数 1

分类专栏： python 文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_43276666/article/details/128116277

版权

1 篇文章 0 订阅

订阅专栏

先到英伟达官网下载TensorRThttps://developer.nvidia.com/nvidia-tensorrt-8x-download（需要登陆英伟达账号）。

注意选择与CUDA对应的版本，下载到电脑后解压。

把bin、include、lib复制到cuda对应文件夹下。复制好后进入到python目录下。

在VS Code的终端切换到上面的路径安装python tensorrt。

pip3 install tensorrt-8.4.3.1-cp310-none-win_amd64.whl

安装完成后可以用一小段代码来测试。

import tensorrt as trt

if __name__ == "__main__":
    print(trt.__version__)
    print("hello trt!!")

输出正常，安装成功！

英伟达官方安装参考https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-zip 。如果需要用到TensorFlow或PyTorch，需要安装uff、graphsurgeon和onnx_graphsurgeon包，这些包都在TensorRT的路径中，安装方法与python tensorrt的相同。