繁体   English   中英

Tensorflow 获取默认设备名称

Tensorflow get the default device name

提示:本站收集StackOverFlow近2千万问答,支持中英文搜索,鼠标放在语句上弹窗显示对应的参考中文或英文, 本站还提供   中文繁体   英文版本   中英对照 版本,有任何建议请联系yoyou2525@163.com。

在 Tensorflow 1.14 中,我尝试使用tf.data.experimental.prefetch_to_device(device=...)将我的数据预取到 GPU。 但我并不总是在 GPU 上进行训练,我经常在 CPU 上进行训练(尤其是在开发过程中)。

有没有办法让当前的默认设备处于使用状态? Tensorflow 要么选择 CPU(当我设置CUDA_VISIBLE_DEVICES=-1时),否则它将选择 GPU,默认值通常有效。

到目前为止,我只能找到一种使用sess.list_devices()列出可见设备的方法,但是必须有一种方法可以查询当前的默认设备,这样我就不必每次都在prefetch_to_device中手动更改它,对吧?

2 个回复

目前没有 API 方法可以按照您所说的进行操作。 最近的

device = 'gpu:0' if tf.test.is_gpu_available() else 'cpu'

是你已经说过的。

我这么认为的原因是分配是在低级别完成的: https://github.com/tensorflow/tensorflow/blob/cf4dbb45ffb4d6ea0dc9c2ecfb514e874092cd16/tensorflow/core/common_runtime/colocation_graph.cc

也许您也可以尝试soft placement

希望能帮助到你。

到目前为止我发现的最佳解决方案是使用tf.test.is_gpu_available

device = 'gpu:0' if tf.test.is_gpu_available() else 'cpu'
1 获取设备名称

如何获取驱动器号的设备名称 示例:如何获取G的设备名称:/ 感谢您的任何帮助 ...

2 TensorFlow 设备名称的详细组成?

什么是设备名称中的“后缀”,例如Allocators 、 Compute和Tensors ,如下所示: 这些是针对给定 CPU 上的进程(或线程)吗? 但是对于第一种没有指定 CPU 的情况呢? 顺便说一句,我已经看到timeline.json (从Timeline序列化)中报告的这些示例。 ...

3 如何获取iOS设备名称

这个问题已经在这里有了答案: 如何获得iPhone的设备名称 8个答案 如何以编程方式获取iOS设备的设备名称,即“鲍勃的iPad”。 该名称在“设置”>“常规”>“关于”>“名称”中可见 ...

4 获取设备名称 php js

是否可以在 php 或 javascript 中获取计算机名称? 如果我使用这个功能,它只能在本地正常工作 ...

5 Amazon FireTV 获取设备名称

我目前正在为 firetv 立方体开发一个 android 应用程序。 我对如何获取亚马逊 fire tv 设备名称有些困惑。 我的意思是设备的“呼叫”名称,例如“Exeu's FireTV Cube”。 只是您可以在设备设置中配置的名称。 有没有办法得到这个? 目前我可以通过以下方式获取一些 ...

6 如何获取完整的设备名称

我在Ubuntu上使用 bettercap gatt package 对本地蓝牙设备进行 LE 扫描。 对于某些设备(主要是 Apple 设备),我得到一个缩短的设备名称,即。 “iPhone”与“Bobs iPhone8”相对。 我意识到广告数据包的空间有限(32 字节),所以有时名称会被截断, ...

7 获取设备名称 php

我知道不可能获得设备名称(例如 PC-Francesco),但我想知道有什么替代解决方案可以获得设备的唯一字符串。 我需要它在登录期间进行比较,并检查设备是否在数据库中设置为被阻止。 我正在考虑合并提供商名称+操作系统+设备类型。 您还有其他更好的解决方案吗? 输入是否可以获得设备的序列号或 mo ...

2020-06-29 09:22:51 0 55   php
8 如何获取DeepLens设备名称

我有3个Deeplens设备运行相同的面部检测模型。 他们将输出发送到某个s3存储桶,该存储桶开始另一个lambda。 但是,我需要一个deeplens才能将输出发送到单独的s3存储桶。 为此,我相信我需要获取deeplens设备名称。 有谁知道我如何从greengrass lambda或一些类似 ...

9 获取电池设备名称

我试图使用大多数是从Microsoft教程获得的代码来获取计算机上所有电池设备的名称,但无法正常工作。 我正在尝试打印所有电池的名称,但是当我运行该程序时,它根本不打印任何内容,但是当我请求一个battery_information结构而不是字符串BatteryDeviceName时,它 ...

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2022 STACKOOM.COM