NVIDIA显卡驱动+CUDA+CUDNN+Tensorflow+Keras之间的关系(含各版本对应关系表)

本文详细介绍了NVIDIA驱动、CUDA、CUDNN与Tensorflow-GPU及Keras之间的关系。驱动决定了CUDA版本,CUDA决定CUDNN和Tensorflow-GPU版本,而Tensorflow-GPU又与Keras有对应关系。通过nvidia-smi等命令可查看相关版本。conda环境下安装的CUDA和CUDNN是系统CUDA的子集,仅服务于特定框架。虚拟环境与系统安装的CUDA库在使用时会根据需要进行调用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

基础介绍

  • Driver

    与键盘驱动、鼠标驱动、打印机驱动没本质区别,硬件统一CPU和外设通信的接口,以驱动程序的形式呈现

    可使用nvidia-smi命令查看,需要注意的是,显卡支持的驱动一直是在更新的

  • CUDACompute Unified Device Architecture,NVIDIA推出的显卡并行计算框架

    tensorflow-gpu的运行需要它的底层支持,它是一个计算框架,抽象层次比驱动高,每个版本的CUDA都是基于一定版本的驱动建立的,所以它对驱动的最低版本是有要求的

  • cuDNNThe NVIDIA CUDA Deep Neural Network library基于CUDA架构的神经网络库

    是专门用于神经网络计算的加速包,可以看作是CUDA的一部分(并行计算框架下的一个包),cuDNN和CUDA的版本也存在对应关系

  • tensorflow-gpu

    tensorflow-gpu的运行需要用到CUDA框架, 特别是cuDNN库,就像CUDA是基于特定版本驱动构建一样,tensorflow-gpu也是调用特定版本的CUDA,二者存在对应关系

  • 关系图

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值