cuda10.2 cudnn7.6 TensorFlow-gpu2.2.0遇到的坑

(纯纯小白的新手历程)
记录一下cuda cudnn TensorFlow遇到的坑
起因是想把一个用cpu跑的TensorFlow改成gpu。

  1. 说是TensorFlow 2.0往上就不用区分cpu gpu了,结果亲测不行,最后还是pip tensorflow-gpu=2.2.0了

  2. cuda的版本,很多教程说让要看先显卡信息,其实这里是最高可支持的,比这个低的也可以,最好不要用最新版本 的。。。显卡信息

  3. cudart64_101
    找不到这个的话,参考一个博主的做法(https://blog.csdn.net/qq_32939413/article/details/105525025),去搜cudart64,应该目录里是有cudart64_102的,直接改名就行。
    在这里插入图片描述

  4. 模型训练错误:Failed to get convolution algorithm,This is probably because cuDNN failed to initialize
    在这个文件
    在这里插入图片描述
    输入
    config = tf.compat.v1.ConfigProto()
    config.gpu_options.allow_growth = True
    tf_keras_backend.set_session(tf.compat.v1.Session(config=config))

基本就是这些坑,弄好之后,就感觉训练速度立马提升很多了

还有几个用到的链接:
1.tensorflow cuda cudnn对应版本
2.anaconda相关清华园镜像

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值