• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    神经网络训练采用gpu设置的方式

    在定义图结构之前不用加入gpu:0,只有在session中计算之前在加入,否则的话会提示不能使用gpu保存模型等问题。

     with
      tf.device(
     '/gpu:0'
      ):
    

    补充:关于应用gpu训练神经网络的注意事项

    对于GPU来说,一定要注意的是,要分别在两个GPU上,或者不同时的在一个GPU上运行train和evaluation的部分,否则限于GPU擅长迭代而不擅长逻辑的特性,会发生OOM(out of memory)

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。

    您可能感兴趣的文章:
    • tensorflow使用指定gpu的方法
    • Pytorch 高效使用GPU的操作
    • 基于Tensorflow使用CPU而不用GPU问题的解决
    • pytorch使用指定GPU训练的实例
    上一篇:解决TensorFlow训练模型及保存数量限制的问题
    下一篇:用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    神经网络训练采用gpu设置的方式 神经网络,训练,采用,gpu,