在兩個GPU上訓練兩個模型

Question

我有兩個模型（theano腳本），我想訓練和評估。
我有兩個GPU可以用來訓練它們。

如何在同一時間在每個GPU上運行模型？

Answer 1

當運行腳本，你可以選擇在您的編程'將與THEANO_FLAGS運行：爲每個GPU

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

變化gpuX（如gpu0，GPU1，GPU2 ...。）