在两个GPU上训练两个模型

时间:2017-06-27 12:51:51

标签: gpu keras theano theano-cuda

我有两个模型(theano脚本)我想训练和评估 我有两个可以用来训练它们的GPU。

如何在每个GPU上同时运行模型?

1 个答案:

答案 0 :(得分:0)

运行脚本时,您可以选择使用THEANO_FLAGS运行程序的位置:

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

为每个gpu更改gpuX(例如.gpu0,gpu1,gpu2 ...)