使用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 FilesNVIDIA GPU Computing ToolkitCUDAv10.0

原文地址:https://www.cnblogs.com/bai2018/p/10468628.html