1. 首页
  2. 编程语言
  3. Python
  4. 神经网络训练采用gpu设置的方式

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

上传者: 2021-07-03 14:56:49上传 PDF文件 93.58 KB 热度 14次

在定义图结构之前不用加入gpu:0,只有在session中计算之前在加入,否则的话会提示不能使用gpu保存模型等问题。对于GPU来说,一定要注意的是,要分别在两个GPU上,或者不同时的在一个GPU上运行train和evaluation的部分,否则限于GPU擅长迭代而不擅长逻辑的特性,会发生OOM

用户评论