Tensorflow не использует GPU - PullRequest
Новая
       21

Tensorflow не использует GPU

0 голосов
/ 04 сентября 2018

Я использую ноутбук 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

Проблема в том, что тензорный поток не использует графический процессор. Я установил библиотеки Cuda 9 и Cudnn и установил тензор потока только после успешной установки cuda. ​​

Я также провел первичные проверки, обозначенные решениями, чтобы выяснить, используется ли графический процессор, и вот результаты: 

    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 . Хотя можно предположить, что это не подходящая мера / метрика, я заметил, что при использовании для игр индикатор показывает использование графического процессора, но во время выполнения кода, похоже, используется только встроенная оперативная память.

enter image description here

Как я могу получить свой код для использования графического процессора?

Подробнее

информация о ноутбуке по этой ссылке

Пожалуйста, сообщите, если требуется дополнительная информация.

...