web-dev-qa-db-fra.com

Keras dans PyCharm n'utilisant pas de GPU

Ces discussions n'ont pas résolu mon problème: Keras n'utilise pas de processeur graphique sur Pycharm ayant Python 3.5 et Tensorflow 1.4

Keras avec backend TensorFlow n'utilisant pas de GPU

J'ai installé Tensorflow et Tensorflow-gpu (v.1.12.0) sur un PC sous Windows 10 doté de la carte graphique GTX 750 Ti. Il prend donc en charge CUDA. J'ai également installé les bibliothèques CUDA Toolkit v10 et cuDNN et lorsque je lance nvcc -V sur une invite de commande, je reçois:

nvcc: Compilateur NVIDIA (R) Cuda ...

J'utilise PyCharm et je n'ai aucun problème à exécuter Keras sur un processeur. Mais il n'utilise pas mon GPU.

Quand je tape

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

ça dit

2018-11-25 10: 47: 57.448275: I tensorflow/core/plateforme/cpu_feature_gaurd.cc: 141] Votre CPU prend en charge les instructions selon lesquelles ce binaire TensorFlow n'a pas été compilé pour utilisation: AVX2

[]

Ce que j'ai essayé:

1) J'ai essayé de désinstaller Tensorflow et Tensorflow-gpu et de réinstaller Tensorflow-gpu comme indiqué dans le fil de discussion ci-dessus. Cela ne fonctionnait pas et mon code ne fonctionnait plus sur le processeur et donnait une erreur concernant Tensorflow. Une fois que j'ai réinstallé Tensorflow, c'était à nouveau OK.

2) J'ai essayé d'utiliser une bibliothèque appelée Theano, qui est mentionnée dans la documentation officielle de Keras ici . Quand j'ajoute les lignes suivantes

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

ça dit

AVERTISSEMENT (theano.configdefaults): g ++ non disponible si vous utilisez conda: 'conda install m2w64-toolchain' C:\Utilisateurs\HOME-PC\PyCharmProjects\env\lib\site-packages\theano\configdefaults.py. =: 560 : UserWarning: DeprecationWarning: il n’existe pas de compilateur c ++. C’est obsolète et avec l’anonymat 0.11, un compilateur c ++ sera obligatoire. warning.warn ("DeprecationWarning: il n'y a pas de compilateur c ++."

3) J'ai essayé d'ajouter ces lignes et rien ne s'est passé.

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

Qu'est-ce que j'oublie ici? Comment dois-je présenter CUDA à PyCharm?

tensorflowkeraspycharmcudnn
4
Alireza

Le problème était avec la version de CUDA. J'avais installé CUDA v10.0 mais Tensorflow semble fonctionner uniquement avec la v9.0. Installé et ça fonctionne comme un charme.

1
Alireza

Ce n’est peut-être pas le cas, mais installer et importer la même bibliothèque peut parfois être déroutant. l'environnement de pycharm. Pour vous assurer d’avoir installé un paquet dans l’environnement de pycharm, 

vous pouvez essayer ce qui suit, à partir de la console python de pycharm

! python -m pip install --upgrade de tensor, keras 

cela installera correctement les paquets, et vous pouvez être sûr que l'installation n'est pas en cause

1
dragonLOLz