查看使用的tensorflow是CPU还是GPU版本
【摘要】 查看使用的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)