环境准备
主要是根据工程环境需要,参考trt文档安装trt
- 1.查看trt适配情况 :链接跳转,注意,不同版本的trt有不同版本的文档,请以最新文档为准。
- 2.根据自己系统情况下载相关包 因为我自己拉的docker镜像是ubuntu16.04,CUDA10的驱动,所以我还需要下载cuDNN和TensorRT。
目录
cuDNN安装
- 去官网下载合适版本的cuDNN ==> 解压 ==> 复制相关文件到系统的CUDA目录
下载完安装包后如下:
root@iZbp120zfnu5353tdx5hq7Z:/data/workspace/docker/trtfile/test# ls
cudnn-1

最低0.47元/天 解锁文章
7902

被折叠的 条评论
为什么被折叠?



