Tensor flow toggle between CPU/GPU

前端 未结 6 1191
情歌与酒
情歌与酒 2021-01-30 13:30

Having installed tensorflow GPU (running on a measly NVIDIA GeForce 950), I would like to compare performance with the CPU.

I am running the tensorFlow MNIST tutorial c

6条回答
  •  误落风尘
    2021-01-30 14:04

    try setting tf.device to cpu:0

    with tf.Session() as sess:
         with tf.device("/cpu:0"):
    

提交回复
热议问题