Posts

Post not yet marked as solved
2 Replies
0 Views
This works for me: with tf.device("/cpu:0"): or with tf.device("/gpu:0"):