[纪念浪费的时间] 安装tensorflow-gpu + cuda + cudnn 之那些踩过的坑

首先一定要确认下载到的tensorflow cuda 和cudnn的版本是正确的

1.首先卸载tensorflow(cpu)版本,(如果有的话),没有就不用了

2.使用anaconda,安装tensorflow-gpu ,输入pip install tensorflow-gpu即可(注意,这里要架梯子才快,此外要以管理员权限打开anaconda prompt)

3.下载cuda,我在北邮人上下载的cuda9.0

4.下载cudnn,注意,一定要下载for cuda9.0的版本

Download cuDNN v7.1.3 (April 17, 2018), for CUDA 9.0

5.安装cuda,安装失败的话,多装几次,安装完毕后一定要重启,否则cuda自动卸载.........exm???

6.添加cuda 的环境变量

7.cudnn的zip解压缩后一定要放在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0下,不要放在CUDA下

8.更新gpu驱动,tensorflow最好是最新版本的1.8.0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值