首页 > 其他 > 详细

在anaconda中查看某个环境中的tensorflow是cpu还是gpu

时间:2021-04-18 22:22:03      阅读:29      评论:0      收藏:0      [点我收藏+]

在anaconda中查看某个环境中的tensorflow是cpu还是gpu

 

>>> import os
>>> from tensorflow.python.client import device_lib
>>> os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
>>>
>>> print(device_lib.list_local_devices())
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5425529995648658955
]
>>>
>>>

 

 

 

 

#############

在anaconda中查看某个环境中的tensorflow是cpu还是gpu

原文:https://www.cnblogs.com/herd/p/14674556.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!