【662】TensorFlow GPU 相关配置

[1] CUDA与cuDNN

[2] Windows10下 tensorflow-gpu 配置

　　需要安装：

• TensorFlow GPU 版本：

  • 与 CPU 版本不同，需要单独安装

  • conda install tensorflow-gpu　　

• CUDA：

  • CUDA (ComputeUnified Device Architecture)，是显卡厂商 NVIDIA 推出的运算平台。 CUDA 是一种由 NVIDIA 推出的通用并行计算架构，该架构使 GPU 能够解决复杂的计算问题。

  • CUDA看作是一个工作台，上面配有很多工具，如锤子、螺丝刀等。

• cuDNN：

  • NVIDIA cuDNN 是用于深度神经网络的 GPU 加速库。它强调性能、易用性和低内存开销。

  • cuDNN是基于CUDA的深度学习GPU加速库，有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具，比如它就是个扳手。但是CUDA这个工作台买来的时候，并没有送扳手。想要在CUDA上运行深度神经网络，就要安装cuDNN，就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作，工作速度相较CPU快很多。