tensorflow如何设置只在cpu上运行
1 回答
手掌心
TA贡献1942条经验 获得超3个赞
with tf.Session() as sess:
with tf.device("/cpu:0"):
后面写你的代码
如果是新手的话，建议安装tensorflow而不是tensorflow-gpu
- 1 回答
- 0 关注
- 2575 浏览
添加回答
举报
0/150
提交
取消