使用tensorflow下的GPU加速神经网络训练过程

下载CUDA8.0，安装

下载cuDNN v5.1安装。放置环境变量等。

其他版本就不装了。不用找其他版本的关系。

使用tensorflow-gpu1.0版本。

使用keras2.0版本。

 

有提示的。

有时候可能需要分配使用空间自动增长：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

英伟达 MX150。华硕FL8000

 

TensorFlow版本：1.1.4

python版本：3.7.3

CUDA版本：10.0

和对应的cudnn版本。

更新地址：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0