Tensorflow-GPU安装壁坑指南

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值