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

tensorflow如何设置只在cpu上运行?

tensorflow如何设置只在cpu上运行?

万千封印 2018-11-26 14:10:25
tensorflow如何设置只在cpu上运行
查看完整描述

1 回答

?
手掌心

TA贡献1942条经验 获得超3个赞

with tf.Session() as sess:
with tf.device("/cpu:0"):
后面写你的代码
如果是新手的话,建议安装tensorflow而不是tensorflow-gpu

查看完整回答
反对 回复 2018-12-10
  • 1 回答
  • 0 关注
  • 2510 浏览

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信