标签: gpu keras theano theano-cuda
我有两个模型(theano脚本)我想训练和评估 我有两个可以用来训练它们的GPU。
如何在每个GPU上同时运行模型?
答案 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 ...)