在两个 GPU 上训练两个模型

Train two models on two GPUs

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

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

当 运行 设置您的脚本时,您可以选择您的程序 运行 与 THEANO_FLAGS:

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

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