conda环境下tensorrt安装

创建conda环境

conda create -n tensorrt python=3.5

下载tensorrt
登录:https://developer.nvidia.com/tensorrt
进入:https://developer.nvidia.com/nvidia-tensorrt-7x-download
在Tar File Install Packages For Linux x86下,选择
TensorRT 7.0.0.11 for Ubuntu 16.04 and CUDA 10.0 tar package
下载获取的文件TensorRT-7.0.0.11.Ubuntu-16.04.x86_64-gnu.cuda-9.0.cudnn7.6.tar.gz

requirements.txt

numpy==1.13.1
pyasn1===0.4.6
six==1.10.0
opencv-python==4.4.0.42
pillow==4.1.1
torch==0.4
importlib-metadata

cuda版本9.0,cudnn版本7.6.5。cudnn版本不对,会出现tenssorrt中调用的函数,在cudnn中未定义错误。
安装


                
### 创建支持 TensorRTConda 虚拟环境 为了在 Ubuntu 上创建能够安装和运行 TensorRTConda 虚拟环境,需遵循特定步骤来确保兼容性和功能正常。这涉及到了解操作系统需求、CUDA 和 cuDNN 版本匹配等问题。 #### 准备工作 确认已安装 Anaconda 或 Miniconda 并更新至最新版本。同时,了解目标平台所需的 CUDA 和 cuDNN 版本对于成功集成 TensorRT 至关重要[^2]。 #### 创建 Python 环境 使用 `conda` 命令行工具指定所需 Python 版本来建立新的虚拟环境: ```shell conda create -n tensorrt_env python=3.8 ``` 激活新建的虚拟环境以便后续操作均在此环境中执行: ```shell conda activate tensorrt_env ``` #### 安装依赖项 根据所选 CUDA 和 cuDNN 版本调整命令参数以适应具体项目需求。这里假设采用 CUDA 10.2 和 cuDNN 8.0 进行说明[^3]: ```shell conda install cudatoolkit=10.2 cudnn=8.0 ``` #### 下载与配置 TensorRT 获取适用于当前系统的 TensorRT 发行版压缩包,并按照官方指导完成本地化设置。此过程通常包括但不限于解压文件、设定必要的环境变量等动作[^4]。 例如,当处理基于 Ubuntu 18.04 LTS 构建的应用场景时,可参照如下指令序列操作: ```shell tar -xvzf TensorRT-7.1.3.4.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn8.0.tar.gz echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/user/TensorRT-7.1.3.4/lib' >> ~/.bashrc source ~/.bashrc ``` 请注意替换 `/home/user/` 部分为实际路径前缀。 #### 将 TensorRT 库引入 Conda 环境 为了让 Python 解释器识别来自外部位置的 TensorRT 组件,建议将位于系统全局区域内的 `tensorrt` 文件夹拷贝到活跃 Conda 环境对应的站点包目录内[^1]。 可以通过下面的方法实现这一点(假定活动环境名称为 `tensorrt_env`): ```shell cp -r /path/to/dist-packages/tensorrt ~/miniconda3/envs/tensorrt_env/lib/python3.8/site-packages/ ``` 再次强调应依据实际情况修正上述命令里的路径部分。 #### 测试安装成果 最后一步是对整个流程的结果加以验证,尝试导入 TensorRT 模块并查看是否有任何异常提示出现: ```python import tensorrt as trt print(trt.__version__) ``` 如果一切顺利,则表明已经成功构建了一个适配 TensorRTConda 工作区。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值