Posts

Post not yet marked as solved
2 Replies
0 Views
execute the code after tf.device("/gpu:0"): for exmaple: """ with tf.device("/cpu:0"): {your code} """ don't forget the indentation before your code