linux判断安装的tensorflow是gpu还是cpu版本

素颜马尾好姑娘i 2021-12-12 16:31 326阅读 0赞

判断代码：

import os
    from tensorflow.python.client import device_lib
    os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
     
    if __name__ == "__main__":
        print(device_lib.list_local_devices())

结果一：CPU

> \[name: "/device:CPU:0"
> 
> device\_type: "CPU"
> 
> memory\_limit: 268435456
> 
> locality \{
> 
> \}
> 
> incarnation: 4723458630420161305
> 
> , name: "/device:XLA\_CPU:0"
> 
> device\_type: "XLA\_CPU"
> 
> memory\_limit: 17179869184
> 
> locality \{
> 
> \}
> 
> incarnation: 11640713468301226911
> 
> physical\_device\_desc: "device: XLA\_CPU device"
> 
> \]

结果二：GPU

> \[name: "/device:CPU:0"
> 
> device\_type: "CPU"
> 
> memory\_limit: 268435456
> 
> locality \{
> 
> \}
> 
> incarnation: 17503920527771229331
> 
> , name: "/device:GPU:0"
> 
> device\_type: "GPU"
> 
> memory\_limit: 191365120
> 
> locality \{
> 
>   bus\_id: 1
> 
>   links \{
> 
>     link \{
> 
>       device\_id: 1
> 
>       type: "StreamExecutor"
> 
>       strength: 1
> 
>     \}
> 
>   \}
> 
> \}
> 
> incarnation: 11078361316836643347
> 
> physical\_device\_desc: "device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:03:00.0, compute capability: 6.1"
> 
> , name: "/device:GPU:1"
> 
> device\_type: "GPU"
> 
> memory\_limit: 6526258381
> 
> locality \{
> 
>   bus\_id: 1
> 
>   links \{
> 
>     link \{
> 
>       type: "StreamExecutor"
> 
>       strength: 1
> 
>     \}
> 
>   \}
> 
> \}
> 
> incarnation: 15928025279328008063
> 
> physical\_device\_desc: "device: 1, name: GeForce GTX 1080 Ti, pci bus id: 0000:04:00.0, compute capability: 6.1"
> 
> \]

发表评论取消回复

表情：

评论列表（有 0 条评论，326人围观）

还没有评论，来说两句吧...

相关阅读

相关 Tensorflow GPU 版本安装教程

非常详细的 Tensorflow GPU 版本安装教程一、安装Anaconda 二、TensorFlow GPU 一、安装Anaconda 这一步比

阳光穿透心脏的1/2处/ 2023年10月09日 18:41/ 0 赞/ 47 阅读

相关【tensorflow】如何安装tensorflow的GPU版本

自从上次尝试复现R-CNN的代码，然后搞崩了tensorflow后，因为网速实在是太卡，安装了几次都已失败告终，下午顺利安装成功，分享一下经验。先看一下自己的CUDA版本：

太过爱你忘了你带给我的痛/ 2023年09月27日 12:18/ 0 赞/ 131 阅读

相关 linux Centos 安装 Tensorflow GPU版本安装教程

![20191009191333910.png][][日萌社][Link 1] [人工智能AI：Keras PyTorch MXNet TensorFlow Paddle

╰半夏微凉°/ 2023年07月13日 04:25/ 0 赞/ 55 阅读

相关查看linux版本是 centos还是ubuntu

> 原文地址，收藏备用 > https://www.php.cn/linux-435827.html 1、lsb\_release -a 如果是想查看你的Linux系统

蔚落/ 2022年11月02日 13:29/ 0 赞/ 277 阅读

相关 Java判断OS是linux还是windows

public class Test \{ public static void main(String aa\[\]) \{ if (System.getProperty("o

本是古典何须时尚/ 2022年07月15日 08:24/ 0 赞/ 219 阅读

相关确认自己的TensorFlow是CPU版本还是GPU版本

这是一个运行TensorFlow操作的类，他封装了操作和Tensor的计算环境那怎么确认我们是安装的那个呢？记录设备指派情况为了获取你的 operations 和

àì夳堔傛蜴生んèń/ 2022年05月09日 02:40/ 0 赞/ 223 阅读

相关 Win10安装anaconda+python3.6+TensorFlow（CPU版本/GPU版本）

1. 下载64位的anaconda（anaconda是一个开源的[Python][]发行版本，其包含了conda、Python等180多个科学包及其依赖项）下载地址：

青旅半醒/ 2022年03月22日 08:09/ 0 赞/ 330 阅读

相关 linux判断安装的tensorflow是gpu还是cpu版本

判断代码： import os from tensorflow.python.client import device_lib os.environ[

素颜马尾好姑娘i/ 2021年12月12日 16:31/ 0 赞/ 327 阅读

相关查看 tensorflow 是 gpu 还是 cpu版本

查看tensorflow 是gpu还是 cpu 在 Python 输入： import os from tensorflow.python.client

我不是女神ヾ/ 2021年12月05日 16:46/ 0 赞/ 278 阅读

相关（最新）win10简单快速安装TensorFlow（CPU版本）and（GPU版本）

\----------------------------希望安装tensorflow不会对你学习tensorflow失去信心-------------------------

本是古典何须时尚/ 2021年08月19日 21:47/ 1 赞/ 859 阅读