conda 安装 tensorrt8

CRISTIANO Xusanduo

于 2021-10-18 11:54:09 发布

阅读量4.6k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： AI 文章标签：神经网络

本文链接：https://blog.csdn.net/shi_xin/article/details/120823725

本文档介绍了如何在conda环境中安装TensorRT8.2，包括推荐的CUDA和cuDNN版本，以及安装过程。首先创建TensorRT专用的conda环境，然后安装CUDA和cuDNN，最后通过pip安装TensorRT并验证安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

conda 安装 tensorrt，目前官网推荐最新的tensorrt8. 最新的tensorrt8，在conda环境下安装，并不复杂。

官网：github

官网推荐环境：

TensorRT GA build

    TensorRT v8.2.0.6

System Packages

    CUDA
        Recommended versions:
        cuda-11.4.x + cuDNN-8.2
        cuda-10.2 + cuDNN-8.2
    GNU make >= v4.1
    cmake >= v3.13
    python >= v3.6.9
    pip >= v19.0
    Essential utilities
        git, pkg-config, wget, zlib

conda 虚拟环境：使用推荐得python3.8环境

conda create -n tensorrt python=3.8

安装cuda及cudnn：

conda install cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

conda install cudnn=8.2.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

安装tensorrt：

也可以参考官网：

最低0.47元/天解锁文章

200万优质内容无限畅学