Tensorflow 与Cuda、cudnn对应关系

Cuda和cudnn是支持NVIDIA支持GPU的两个库,分别用于高性能计算和深度神经网络计算的支持。

Cuda(Compute Unified Device Architecture),是NVIDIA支持GPU的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。可以看作使能GPU的驱动程序或算法SDK。

cudnn是基于Cuda的深度学习GPU加速库,有了它可以在GPU上完成高效的深度学习计算。

Cuda历史版本下载地址:

https://developer.nvidia.com/cuda-toolkit-archive

cudnn历史版本下载地址:

https://developer.nvidia.com/rdp/cudnn-archive

其中在cudnn历史版本下载页面可以看到与Cuda的版本对应关系。

tensorflow版本及支持的对应Cuda和cudnn版本

https://tensorflow.google.cn/install/source_windows?hl=cn

来源:Dnlcwyzct

物联沃分享整理
物联沃-IOTWORD物联网 » Cuda与cudnn对应关系

发表评论