安装CUDA使用GPU训练模型——python3.6,tensorflow-gpu 1.15.0,cuda 10.0
本节主要讲述在模型训练时利用gpu对训练进行加速首先我们需要知道gpu不是我们想调用就可以直接调用的,我们需要安装一个cuda工具包以及其对应的cudnn(cuDNN是用于配置深度学习使用),当我们安装好这两个时才能利用机器学习来进行训练,其次我们的gpu驱动要足够新,他会对版本更低的cuda工具包进行兼容,但更高的不行(不需要降级gpu驱动),最后便是最复杂的问题:版本对应,可以去官网,我使用的