[1] CUDA与cuDNN
需要安装:
-
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快很多。
-