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中未定义错误。
安装


                
<think> 我们参考了多个引用,其中引用[2]提供了在conda虚拟环境中使用pip安装TensorRT的方法,并验证安装成功的步骤。 另外,引用[1]提到了一种方法:将系统Python的dist-packages中的tensorrt复制到虚拟环境的site-packages中,但这种方法可能不如使用pip安装方便。 因此,我们推荐使用pip安装方法,具体步骤如下: 1. 激活conda虚拟环境。 2. 安装nvidia-pyindex,这是NVIDIA的Python包索引。 3. 使用pip安装nvidia-tensorrt。 验证安装: 在Python环境中导入tensorrt并打印版本,同时尝试创建一个Builder对象(需要确保有可用的CUDA环境)。 注意:安装TensorRT需要与系统中已安装的CUDA版本匹配。请提前确认CUDA版本,并选择对应的TensorRT版本。 详细步骤: 步骤1:激活conda虚拟环境 conda activate your_env_name 步骤2:安装nvidia-pyindex python -m pip install nvidia-pyindex 步骤3:安装TensorRT python -m pip install --upgrade nvidia-tensorrt 步骤4:验证安装 启动Python,执行: import tensorrt print(tensorrt.__version__) assert tensorrt.Builder(tensorrt.Logger()) 如果以上步骤没有报错,则安装成功。 另外,引用[3][4][5]提到在Windows下安装需要注意CUDA版本对应,以及下载对应的TensorRT包。但上述pip方法在Linux和Windows上应该都适用(注意Windows下需要对应的CUDA环境变量设置,并且确保pip安装TensorRT与CUDA版本匹配)。 如果遇到问题,可以检查以下事项: - 确保虚拟环境中已经安装了与TensorRT兼容的CUDA Toolkit(通常需要系统安装CUDA,并在虚拟环境中通过conda安装cudatoolkit,或者使用系统级CUDA并设置环境变量)。 - 如果使用系统CUDA,请确保虚拟环境中能够找到CUDA(例如设置LD_LIBRARY_PATH(Linux)或PATH(Windows)包含CUDA的库路径)。 - 如果pip安装TensorRT版本与CUDA版本不匹配,可能需要指定版本,例如:`pip install nvidia-tensorrt==8.6.1`(具体版本根据需求)。 注意:TensorRT安装可能还需要cuDNN的支持,同样需要确保cuDNN的版本与TensorRT和CUDA匹配。 如果上述方法不行,也可以尝试从NVIDIA官网下载TensorRT的压缩包(.tar.gz或Windows的zip),然后按照官方文档进行安装,并将相关的库路径添加到环境变量中。但这种方法相对复杂,且需要手动将Python包复制到虚拟环境(如引用[1]所述)。 由于我们推荐使用pip安装,所以先尝试上述步骤。 生成相关问题: </think> 在conda虚拟环境安装TensorRT的推荐方法是使用pip安装,以下是详细步骤: ### 安装方法 1. **激活虚拟环境** ```bash conda activate your_env_name # 替换为你的环境名 ``` 2. **安装依赖索引** ```bash python -m pip install nvidia-pyindex # 添加NVIDIA官方仓库 ``` 3. **安装TensorRT** ```bash python -m pip install --upgrade nvidia-tensorrt ``` > 注:自动安装匹配的CUDA依赖(如cuDNN/CUDA Runtime) 4. **验证安装** ```python >>> import tensorrt >>> print(tensorrt.__version__) # 输出版本号(如8.6.1) >>> assert tensorrt.Builder(tensorrt.Logger()) # 检查运行时环境 ``` ### 注意事项 1. **CUDA版本匹配** - 需确保系统CUDA版本与TensorRT兼容([版本对应表](https://docs.nvidia.com/deeplearning/tensorrt/support-matrix/index.html)) - 检查CUDA版本:`nvcc --version` 2. **Windows特殊配置** - 需手动添加TensorRT库路径到环境变量: ```powershell set PATH=%PATH%;C:\TensorRT\lib # 替换为实际解压路径[^3][^5] ``` 3. **备选方案(不推荐)** 若pip安装失败,可尝试复制系统包: ```bash cp -r /usr/lib/python3.8/dist-packages/tensorrt* $CONDA_PREFIX/lib/python3.9/site-packages/ ``` > 需注意Python版本一致性,可能引发兼容性问题[^1] ### 常见问题解决 - **导入错误**:检查`LD_LIBRARY_PATH`(Linux)或`PATH`(Windows)是否包含CUDA/TensorRT库路径 - **版本冲突**:使用`conda list`检查是否存在冲突包(如旧版tensorflow-gpu) - **安装特定版本**:`pip install nvidia-tensorrt==8.6.1`
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值