Керасы в PyCharm не используют GPU - PullRequest
0 голосов
/ 25 ноября 2018

Эти темы не решили мою проблему: Keras не использует GPU на Pycharm с Python 3.5 и Tensorflow 1.4

Keras с бэкэндом TensorFlow, не использующим GPU

Я установил Tensorflow и Tensorflow-gpu (v.1.12.0) на моем компьютере с Windows 10 и графической картой GTX 750 Ti, поэтому он поддерживает CUDA.Я также установил библиотеки CUDA Toolkit v10 и cuDNN, и когда я запускаю nvcc -V в командной строке, я получаю:

nvcc: компилятор NVIDIA Cuda ...

Я использую PyCharm, и у меня нет проблем с запуском Keras на CPU.Но он не использует мой графический процессор.

Когда я набираю

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

, он говорит

2018-11-25 10: 47: 57.448275: Itenorflow / core / platform / cpu_feature_gaurd.cc: 141] Ваш процессор поддерживает инструкции, которые этот двоичный файл TensorFlow не был скомпилирован для использования: AVX2

[]

Что у меня естьпопробовал:

1) Я попытался удалить Tensorflow и Tensorflow-gpu и переустановить Tensorflow-gpu, как указано выше.Не работал, и мой код больше не работал на CPU и выдал ошибку относительно Tensorflow.Как только я переустановил Tensorflow, все снова было в порядке.

2) Я попытался использовать библиотеку Theano, которая упоминается в официальной документации Keras здесь .Когда я добавляю следующие строки

import theano
theano.config.device='gpu'
theano.config.floatX='float32'

, это говорит

ПРЕДУПРЕЖДЕНИЕ (theano.configdefaults): g ++ недоступен, если используется conda: 'conda install m2w64-toolchain' C:\ Users \ HOME-PC \ PyCharmProjects \ env \ lib \ site-packages \ theano \ configdefaults.py. =: 560: UserWarning: DeprecationWarning: нет компилятора c ++. Это устарело, а с theano 0.11 компилятор c ++ будет обязательным,warning.warn («DeprecationWarning: нет компилятора c ++».

3) Я попытался добавить эти строки, но ничего не произошло.

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0" 

Чего мне здесь не хватает?Как мне представить CUDA для PyCharm?

Ответы [ 2 ]

0 голосов
/ 18 декабря 2018

Проблема была с версией CUDA.Я установил CUDA v10.0, но Tensorflow работает только с v9.0.Установил его и он работает как шарм.

0 голосов
/ 01 декабря 2018

Возможно, дело не в этом, но установка и импорт одного и того же библиотекаря может иногда приводить в замешательство.Более ясно, я предполагаю, что ваша среда Pycharm отличается от среды Python по умолчанию, вы устанавливаете пакеты в среде по умолчанию и импортируете из среды Pycharm.Чтобы убедиться, что у вас установлен пакет в среде pycharm,

, вы можете попробовать следующее с консоли pycharm python

! Python -m pip install --upgrade tenorflow, keras

это должно правильно установить пакеты, и вы можете быть уверены, что установка не виновата

...