如何使用多个 GPU 对 Tensorflow 进行单独训练?

我浏览了许多关于在 TensorFlow 中使用多个 GPU 的文章和帖子。它帮助我更多地了解“如何使用并行 GPU 来训练 NN(神经网络)”here。但我有一个不同的问题。是否可以使用单独的 GPU 同时训练不同的神经网络?

更多详情:

我有神经网络 ABGPU1GPU2。我想同时在 A 上训练 GPU1 NN 和 B 上的 GPU2 NN。可能吗?

luofeng0710 回答:如何使用多个 GPU 对 Tensorflow 进行单独训练?

我建议使用两个单独的 Python 脚本来训练两个网络,例如 trainA.pytrainB.py

trainA.py 的前两行中,您选择首选 GPU。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

对于 trainB.py,您选择另一个 GPU:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

现在您应该可以同时运行两个训练脚本了。

本文链接:https://www.f2er.com/1074606.html

大家都在问