在两个 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...)
我有两个模型(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...)