CODEWITHSUNDEEP
gpukerastheanotheano-cuda
circuitdesigner5172
circuitdesigner5172

Reputation: 125

Train two models on two GPUs

I have two models (theano scripts) I want to train and evaluate.
I have two GPUs I can use to train them.

How can I run a model on each GPU at the same time?

Upvotes: 2

Views: 290

Answers (1)

maz
maz

Reputation: 1980

When running your script you can choose where your progam will run with THEANO_FLAGS:

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

change gpuX for each gpu (eg. gpu0,gpu1,gpu2...)

Upvotes: 1

Related Questions