tensorflow如何設(shè)置只在cpu上運(yùn)行
1 回答
手掌心
TA貢獻(xiàn)1942條經(jīng)驗(yàn) 獲得超3個(gè)贊
with tf.Session() as sess:
with tf.device("/cpu:0"):
后面寫你的代碼
如果是新手的話，建議安裝tensorflow而不是tensorflow-gpu
- 1 回答
- 0 關(guān)注
- 2603 瀏覽
添加回答
舉報(bào)
0/150
提交
取消