Tensorflow-GPU的具体安装步骤请参考知乎大佬文章:点这里
1.安装tensorflow-gpu前先确定好你电脑对应的CUDA和cuDNN版本
这里我的建议是,不要安装最新的CUDA(别问我为什么,一般和你的电脑不兼容),比如我的显卡GTX1050对应的CUDA为10.2,但是我还是装了CUDA10.0才正常运行代码。选择比你电脑显卡对应的CUDA低两个版本的CUDA最好。
至于cuDNN,这个很好搞,自行百度CUDA对应的cuDNN版本,比如CUDA10.0 对应cuDNN2.6.0 (注意可以有多个cuDNN与同一个版本的CUDA对应,都可以使用)
2.选择tensorflow-gpu版本以及与其兼容的keras版本
Tensorflow和CUDA及cuDNN的对应:点这里
Tensorflow-gpu和keras-gpu版本的对应:点这里
这样对应着下载,一般不会再有什么问题了
这里提醒一下,不用非得一一对应,举个例子,我的CUDA为10.0,但我装的tensorflow-gpu为1.10.0 依然可以正常使用,而不用非得是1.13.0版本
3.CUDA安装
安装CUDA前要现在电脑上下载好 Visio Stdio2017 否则是装不上CUDA的
Visio Stdio2017下载地址:百度网盘 提取码: xmcc