基础介绍
-
Driver
与键盘驱动、鼠标驱动、打印机驱动没本质区别,硬件统一CPU和外设通信的接口,以驱动程序的形式呈现
可使用nvidia-smi命令查看,需要注意的是,显卡支持的驱动一直是在更新的
-
CUDA(Compute Unified Device Architecture,NVIDIA推出的显卡并行计算框架)
tensorflow-gpu的运行需要它的底层支持,它是一个计算框架,抽象层次比驱动高,每个版本的CUDA都是基于一定版本的驱动建立的,所以它对驱动的最低版本是有要求的
-
cuDNN(The NVIDIA CUDA Deep Neural Network library,基于CUDA架构的神经网络库)
是专门用于神经网络计算的加速包,可以看作是CUDA的一部分(并行计算框架下的一个包),cuDNN和CUDA的版本也存在对应关系
-
tensorflow-gpu
tensorflow-gpu的运行需要用到CUDA框架, 特别是cuDNN库,就像CUDA是基于特定版本驱动构建一样,tensorflow-gpu也是调用特定版本的CUDA,二者存在对应关系
-
关系图