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