欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 财经 > 产业 > Ubuntu 安装CUDA, cuDNN, TensorRT(草稿)

Ubuntu 安装CUDA, cuDNN, TensorRT(草稿)

2025/2/23 14:05:11 来源:https://blog.csdn.net/qq_39779233/article/details/140332904  浏览:    关键词:Ubuntu 安装CUDA, cuDNN, TensorRT(草稿)

文章目录

  • 写在前面
  • 一、CUDA, cuDNN, TensorRT 三个库的版本的确定
  • 二、解决方法
  • 参考链接

写在前面

自己的测试环境:
Ubuntu20.04,

本文安装的版本:
cuda_11.1.0;cuDNN-8.2.1;TensorRT-8.2.4.2

一、CUDA, cuDNN, TensorRT 三个库的版本的确定

参考如下链接:
[1] initative. CUDA安装&cuDNN、TensorRT版本匹配 [EB/OL]. https://blog.csdn.net/initative/article/details/138213992, 2024-05-14/2024-07-10.

二、解决方法

cudnn:

sudo cp cudnn-11.4-linux-x64-v8.2.4.15/cuda/include/cudnn*.h  /usr/local/cuda-11.4/include
sudo cp cudnn-11.4-linux-x64-v8.2.4.15/cuda/lib64/libcudnn*  /usr/local/cuda-11.4/lib64
sudo cp -P  cudnn-11.4-linux-x64-v8.2.4.15/cuda/lib64/libcudnn*  /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn*.h /usr/local/cuda-11.4/lib64/libcudnn*

在这里插入图片描述

在这里插入图片描述

Tensorrt

wong@wong-laptop:~/Datum/software_package/NVIDIA/TensorRT-8.2.4.2.Linux.x86_64-gnu.cuda-11.4.cudnn8.2/TensorRT-8.2.4.2$ sudo cp -r  -P ./lib/*  /usr/local/TensorRT-8.2.4.2/lib/
wong@wong-laptop:~/Datum/software_package/NVIDIA/TensorRT-8.2.4.2.Linux.x86_64-gnu.cuda-11.4.cudnn8.2/TensorRT-8.2.4.2$ sudo cp -r  -P ./bin/*  /usr/local/TensorRT-8.2.4.2/bin/

现在只有 include 未复制到 /usr/local/TensorRT-8.2.4.2/include/ 下.

参考链接

[1] wangchengjue29. CUDA安装&cuDNN、TensorRT版本匹配 [EB/OL]. https://blog.csdn.net/initative/article/details/138213992, 2024-05-14/ 2024-07-11.

[1] (目前主要参考这个)秋山丶雪绪. 从零开始 TensorRT(1)环境篇:下载、安装、测试 [EB/OL]. https://blog.csdn.net/weixin_43605641/article/details/126744446, 2024-05-11/2024-07-08.
[2] 星光2020. CUDA和cuDNN各版本下载及版本对应关系 [EB/OL]. https://blog.csdn.net/mbdong/article/details/121769951, 2021-12-07/2024-07-08.
[3] stivory.Tensorrt安装及使用(python版本) [EB/OL]. https://blog.csdn.net/stivory/article/details/135847529, 2024-02-28/2024-07-08.
[4] nvidia. cuDNN Archive [EB/OL]. https://developer.nvidia.com/rdp/cudnn-archive, xxxx-xx-xx/2024-07-08.
[5] nvidia. NVIDIA TensorRT 8.x Download [EB/OL]. https://developer.nvidia.com/nvidia-tensorrt-8x-download, xxxx-xx-xx/2024-07-08.

[6] 【深度学习】显卡驱动, cuda, cudnn的关系与版本对应问题 https://blog.csdn.net/All_In_gzx_cc/article/details/126126297

[7] https://docs.nvidia.com/deeplearning/tensorrt/archives/index.html#overview
[8] https://docs.nvidia.com/deeplearning/tensorrt/release-notes/index.html#rel_8-0-3

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词