查看使用的tensorflow是CPU还是GPU版本

举报
山海之光 发表于 2019/08/16 15:55:08 2019/08/16
【摘要】 查看使用的tensorflow是CPU还是GPU版本

有两种方法:

(1)使用pip list查看

这种方法仅适用于你的机器只安装了CPU版或只安装了GPU版的情况,如果pip list的输出结果中只看到tensorflow-gpu,说明安装的是GPU版本,如果只看到tensorflow,说明安装的是CPU版本。


(2)使用device_lib查看

如果你的机器同时安装了tensorflow的CPU和GPU版本,具体跑代码的时候搞不清楚用的是哪个,那么可以使用如下代码查看版本:

from tensorflow.python.client import device_lib

is_gpu_version = False

devices_info = device_lib.list_local_devices()

for device in devices_info:

    if 'GPU' == device.device_type:

        is_gpu_version = True

        break

print(is_gpu_version)


tf还有其他版本号可以查看,不过一般用不上:

tf.__version__,tf的版本号

tf.__git_version__,暂不清楚作用,应该是官方的git版本号,其值是“v1.8.0-0-g93bc2e2072”的形式

tf.__compiler_version__,暂不清楚作用,其值是“4.8.4”的形式

这些版本号的初始化是在import tensorflow as tf的时候发生的,可以使用tf.__file__找到tf的安装目录,打开该目录下的__init__.py,可以发现版本号都在这里导入了。

额外提一下,tensorflow安装目录下有_pywrap_tensorflow_internal.lib和_pywrap_tensorflow_internal.pyd,应该是C编译的库和cpython拓展库,加载CUDA和cuDNN的过程就是在这两个库中定义的。


【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。