Tensorrt8.6.1安装

环境配置:Tensorrt8.6.1 ,cuda11.6, cudnn8.6.0, torch1.13.1+cu116, torchvision0.14.1+cu116

1.先把cuda11.6, cudnn8.6.0, torch1.13.1+cu116, torchvision0.14.1+cu116这几个安装完之后

2.下载Tensorrt8.6.1,复制移动以下文件

将include中头文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include\
将lib中所有lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\lib\x64\
将lib中所有dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin\
 

3.去英伟达官网找到对应的zlibwapi.dll(依据电脑版本下载X64、X86版本的)文件,地址如下:Installation Guide - NVIDIA Docs,并将该dll拷贝到C:\Windows\System32文件夹下

4.在Tensorrt8.6.1/python里面有几个库,在cmd用pip install 库文件名。  来安装这些依赖库(python版本是3.10,所以安装对应的这三个库)

Tensorrt8.6.1安装_第1张图片

5.安装cuda-python库

pip install cuda-python

6.在cmd运行tensorrt测试案例:

Tensorrt8.6.1安装_第2张图片

Tensorrt8.6.1安装_第3张图片

7.运行过程中会显示少一些库,使用pip安装对应的库就OK了。运行成功如图所示

Tensorrt8.6.1安装_第4张图片

你可能感兴趣的:(python,tensorrt,pytorch,conda)