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