使用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

猜你喜欢

转载自www.cnblogs.com/bai2018/p/10468628.html