win10安装TensorRT

网友投稿 1574 2022-05-28

2.1 下载TensorRT

版本使用TensorRT 7.2.3 for Windows,CUDA版本是11.0,

win10安装TensorRT

-:NVIDIA TensorRT 7.x Download | NVIDIA Developer

文件大小约500M,

2.2 配置环境变量

将下载好的文件赋值到C盘根目录,然后将“C:\TensorRT-7.2.3.4\lib”添加环境变量。如下图:

2.3 安装必要的whl文件

在shell里面安装下面三个whl文件。执行命令:

pip install nvidia-pyindex pip install graphsurgeon pip install uff

1

2

3

2.4 复制dll文件到cuda安装目录

复制到:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin,如下图:

到这里就安装完成了,下面我们测试一下。

2.5 测试

使用VS2017,打开C:\TensorRT-7.2.3.4\samples\sampleMNIST的sample_mnist.sln,如图:

然后右键sample_mnist项目,右键选择属性。然后再属性页选择“VC++目录——可执行文件目录”,打开“可执行文件目录”,将“C:\TensorRT-7.2.3.4\lib”添加进去,然后点击“确定”。

选择“C/C++——附加包含目录,点进去添加“C:\TensorRT-7.2.3.4\bin”,然后点击“确定”。

右键“sample_mnist”工程,然后选择“重新生成”。

看到如下信息,则表明生成成功。

生成后的文件如下图:

然后执行在data-》mnist下面运行“download_pgms.py”文件,下载数据集。

然后执行调试。如下图

出现下图则表明成功。

Windows

版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。

上一篇:scratch蝙蝠冲关游戏 电子学会图形化scratch编程等级考试二级真题答案解析2019-5
下一篇:一文带你了解Redis持久化完整版本
相关文章