问题描述
我浏览了许多关于在 TensorFlow 中使用多个 GPU 的文章和帖子。它帮助我更多地了解“如何使用并行 GPU 来训练 NN(神经网络)”here。但我有一个不同的问题。是否可以使用单独的 GPU 同时训练不同的神经网络?
更多详情:
我有神经网络 A
、B
和 GPU1
、GPU2
。我想同时在 A
上训练 GPU1
NN 和 B
上的 GPU2
NN。可能吗?
解决方法
我建议使用两个单独的 Python 脚本来训练两个网络,例如 trainA.py
和 trainB.py
。
在 trainA.py
的前两行中,您选择首选 GPU。
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
对于 trainB.py
,您选择另一个 GPU:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
现在您应该可以同时运行两个训练脚本了。