代码之家 › 专栏 › 技术社区 › Guillaume Chevalier

如何在同一台计算机上的不同GPU上并行运行多个TensorFlow实例?

tensorflow python-3.x python

Guillaume Chevalier · 技术社区 · 7 年前

假设我在同一台机器上同时启动以下命令来训练许多TensorFlow模型:

让我们假装在里面 launch_training.py ,将创建TensorFlow图形和会话,并使用以下上下文: with tf.device('/gpu:0'): 以及 0 被适当的 --gpu 索引参数)。

这行吗?如果没有,我需要采取哪些步骤来实现这一点?在租GPU之前我想知道这个。

1 回复 | 直到 7 年前

Colonder 7 年前

您必须使用指定的GPU设备 with tf.device('gpu:N') 在哪里? N 是设备索引。阅读 https://www.tensorflow.org/programmers_guide/using_gpu 和 https://github.com/carla-simulator/carla/issues/116 第一

我认为您混淆了在不同的GPU上多次运行相同的脚本,以及使用多个GPU运行一个脚本。在前一种情况下,阅读TensorFlow指南中的“在多个GPU系统上使用单个GPU”一节,对于后一节“使用多个GPU”。

推荐文章

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

4 月前

Cam · Pandas列表日期到日期时间

4 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

4 月前

jjkennedy · Pandas文本文件导入:当每个文件中存在多个表时,自动选择1个表

4 月前

LMC · Numpy数组布尔索引以获取包含元素

4 月前

vr8ce · 非成对标记中特定字符的正则表达式

5 月前

Kernel · 如果指定了crs参数,shapefile的geopandas.read_file将出错

5 月前

ShaAnder · 为什么sqllachemy返回的是类而不是字符串

5 月前

sixtytrees · detectron2软件包未安装(没有名为“torch”的模块),但我安装了torch

5 月前

Pernoctador · Python映射可以复制吗?我需要参考地图

5 月前