Как заставить мой код Spyder работать на GPU вместо процессора на Ubuntu?

Question

Я работаю над Spyder, чтобы создать модель глубокого обучения на машине с графическим процессором. Я обнаружил, что я работаю на процессоре, и мой код работает долго. Сначала я скачал GPU tenorflow, но не знаю, как это сделать. начать работать на GPU. Как заставить мой код Spyder работать на GPU вместо процессора на Ubuntu?

Answer 1

Вы выбираете процессор, явно указав устройство следующим образом:

with tf.device("cpu"):
    with tf.Session() as sess:
         your code