为了账号安全,请及时绑定邮箱和手机立即绑定

如何让 Keras 在特定 GPU 上训练模型?

如何让 Keras 在特定 GPU 上训练模型?

哔哔one 2021-09-01 14:49:12
我所在的机构有一台带有 2 个 GPU 的共享服务器。假设有两个团队成员都想同时训练一个模型,那么他们如何让 Keras 在特定的 GPU 上训练他们的模型以避免资源冲突?理想情况下,Keras 应该找出当前哪个 GPU 正在忙于训练模型,然后使用另一个 GPU 来训练另一个模型。然而,情况似乎并非如此。似乎默认情况下,Keras 只使用第一个 GPU(因为Volatile GPU-Util第二个 GPU 的 总是 0%)。
查看完整描述

3 回答

  • 3 回答
  • 0 关注
  • 233 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号