一、Cuda配置
根据电脑显卡选择合适的驱动和cuda版本,历代CUDA版本的下载地址如下:
cuda历史版本下载地址(https://developer.nvidia.com/cuda-toolkit-archive)
cudnn下载地址(https://developer.nvidia.com/rdp/cudnn-archive)
CUDA及cudnn的安装教程可参考博客:https://www.cnblogs.com/arxive/p/11198420.html
笔者安装的是10.1版本的CUDA。
二、TensorRt的安装
2.1 先在官网下载TensorRt
TensorRt官网传送门:https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html
PS:在官网上下载需要注册登录NVIDIA账号。
2.2 将下载好的TensorRt安装包解压,如下图所示:
2.3 在系统环境变量中加入lib的路径,如下图所示:
2.4 将解压之后的dll文件复制到CUDA的安装目录bin/下(CUDA默认安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\xxx)
2.5 安装完毕,进行测试
进入TensorRt的sample文件夹,用VS打开MNIST的测试样例,先对VS进行配置,如下图:
PS:在一步可能会出现编译报错,一般都是由于include文件路径及lib文件路径错误所致,请注意检查。
编译完成之后在tensorRt的bin目录下即可点击exe运行测试样例,如图:
三、附录
介绍TensorRt相关理论知识的博客:https://arleyzhang.github.io/articles/7f4b25ce/
“本站所有文章均为原创,欢迎转载,请注明文章出处:https://blog.csdn.net/ll15982534415/article/details/107516264。百度和各类采集站皆不可信,搜索请谨慎鉴别。技术类文章一般都有时效性,本人习惯不定期对自己的博文进行修正和更新,因此请访问出处以查看本文的最新版本。”
原版:2020/9/7