Tensorflow-GPU安装壁坑指南(持续更新)

旧街凉风 提交于 2020-01-30 05:53:29

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版本

待续… …

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!