Tensorflow不使用GPU

时间:2018-09-04 03:47:25

标签: python tensorflow gpu nvidia

我正在使用MSI GL63 8RD-455IN 2018笔记本电脑,具有以下硬件和软件功能:

    - NVidia GTX 1050 ti
    - Driver version - Nvidia 398.92
    - OS : Win 10/64
    - keras version : 2.2.2
    - python : 3.6.5
    - tensorflow version : 1.10.0 - gpu

问题是,tensorflow没有使用GPU。我安装了Cuda 9和Cudnn库,并且仅在成功安装cuda之后才安装tensorflow。

我还进行了以解决方案表示的主要检查,以查看是否正在使用GPU,结果如下:

    print(device_lib.list_local_devices())

    [name: "/device:CPU:0"
    device_type: "CPU"
    memory_limit: 268435456
    locality {
    }
    incarnation: 2797024620588530864
    , name: "/device:GPU:0"
    device_type: "GPU"
    memory_limit: 3168377241
    locality {
    bus_id: 1
    links {
    }
    }
    incarnation: 6850620683865695635
    physical_device_desc: "device: 0, name: GeForce GTX 1050 Ti, pci bus id: 
    0000:01:00.0, compute capability: 6.1"
    ]

我怀疑的原因是来自 MSI Dragon Center 的GPU使用率指标。尽管可能有人提出这不是适当的度量标准,但我注意到,在用于游戏时,该指示器显示了GPU的使用,但是在运行代码时,似乎仅使用了内置RAM。

enter image description here

如何获取我的代码以使用GPU?

更多信息

link

中的笔记本电脑信息

请告知是否需要更多信息。

0 个答案:

没有答案