栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

我如何检查keras是否使用tensorflow的gpu版本?

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

我如何检查keras是否使用tensorflow的gpu版本?

您正在使用GPU版本。您可以列出可用的tensorflow设备(也请检查此问题):

from tensorflow.python.client import device_libprint(device_lib.list_local_devices()) # list of DeviceAttributes

编辑:

使用tensorflow> =1.4时,您可以运行以下功能:

import tensorflow as tftf.test.is_gpu_available() # True/False# Or only check for gpu's with cuda supporttf.test.is_gpu_available(cuda_only=True)

编辑2:

上面的功能在中已弃用

tensorflow > 2.1
。相反,您应该使用以下功能:

import tensorflow as tftf.config.list_physical_devices('GPU')

注意:

在您的情况下,cpu和gpu均可用,如果使用tensorflow的cpu版本,则不会列出gpu。在您的情况下,无需设置tensorflow设备(

withtf.device("..")
),tensorflow将自动选择您的GPU!

此外,您

sudo pip3 list
清楚地表明您正在使用tensorflow-gpu。如果您拥有tensoflow
cpu版本,则名称将类似于
tensorflow(1.1.0)

检查此问题以获取有关警告的信息。



转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/400115.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号