Former deux modèles sur deux GPU

Question

J'ai deux modèles (scripts theano) que je veux former et évaluer.
J'ai deux GPU que je peux utiliser pour les former.

Comment puis-je exécuter un modèle sur chaque GPU en même temps?

Answer 1

Lors de l'exécution de votre script, vous pouvez choisir où votre progamme fonctionnera avec THEANO_FLAGS:

THEANO_FLAGS='device=gpu0' python script_1.py 
THEANO_FLAGS='device=gpu1' python script_2.py 

changement gpuX pour chaque gpu (par exemple gpu0, GPU1, GPU2 ....)