Czy mogę uruchomić model Keras na GPU?

Question

Używam modelu Keras, z terminem przesyłania wynoszącym 36 godzin, jeśli wyszkolę swój model na procesorze zajmie to około 50 godzin, czy istnieje sposób na uruchomienie Keras na GPU?

Używam backendu Tensorflow i uruchamiam go na moim notebooku Jupyter, bez zainstalowanej anakondy.

Answer 1

Tak, można uruchamiać modele Keras na GPU. Kilka rzeczy musisz najpierw sprawdzić.

1. system ma GPU (Nvidia. Jako AMD jeszcze nie działa)
2. Została zainstalowana wersja GPU tensorflow
3. Masz zainstalowany CUDA installation instructions
4. Sprawdź, czy tensorflow działa z GPU check if GPU is working

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

OR

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

wyjście będzie coś takiego:

[ 
    name: "/cpu:0"device_type: "CPU", 
    name: "/gpu:0"device_type: "GPU" 
] 

Gdy to wszystko odbywa się model będzie działał na GPU jeśli włączysz GPU. Wszystkiego najlepszego.

Answer 2

Pewnie. Przypuszczam, że zainstalowałeś już TensorFlow dla GPU.

Musisz dodać następujący blok po zaimportowaniu keras. Pracuję na komputerze, który ma 56 rdzeń procesora i gpu.

import keras 

config = tf.ConfigProto(device_count = {'GPU': 1 , 'CPU': 56}) 
sess = tf.Session(config=config) 
keras.backend.set_session(sess) 

Oczywiście to użycie wymusza maksymalne limity urządzeń. Można zmniejszyć wartości zużycia procesora i gpu.