找到张量流中检测到的设备数量 [英] find number of detected devices in tensorflow

查看:347
本文介绍了找到张量流中检测到的设备数量的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

由于标题已经说明,我想了解tensorflow能够检测或使用的设备数量(cpus或gpus)。

As the title already says I want to find out the number of devices (cpus or gpus) that tensorflow is able to detect or make use of.

原因对于这个问题是因为我在一个有多个cpus的机器上运行keras,但在运行时只使用一个CPU。

The reason for this question is because I'm running keras with tf backend on a machine with multiple cpus, but it uses only a single CPU at runtime.

推荐答案

查看这个关于CPU缩放的错误报告; https://github.com/tensorflow/tensorflow/issues/583

据推测,在2016年5月初发布的更改ab02c5解决。尝试从源构建tensorflow以获取所有最新的更新,这很可能不是在稳定版本。

It was supposedly resolved with change ab02c5, issued at the beginning of May 2016. Try building tensorflow from source to get all the latest updates, which are most likely not in the stable version.

这篇关于找到张量流中检测到的设备数量的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆