下载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