Comment utiliser TensorFlow GPU?

Comment utiliser TensorFlow GPU version au lieu de CPU version en Python 3.6 x64?

import tensorflow as tf

Python est à l'aide de mon CPU pour les calculs.

Je peux le remarquer parce que j'ai une erreur:

Votre CPU prend en charge les instructions que ce TensorFlow binaire n'a pas été
compilé à utiliser: AVX2

J'ai installé tensorflow et tensorflow-gpu.

Comment changer de GPU version?

  • Avez-vous essayé de désinstaller tensorflow et juste garder les tensorflow-gpu installé?
  • Essayez de télécharger à CUDA et l'installation de la GPU de la version.
  • C'est juste un avertissement, si vous avez un GPU NVIDIA, Tensorflow-gpu va automatiquement utiliser. Pour en savoir plus et comment faire pour désactiver l'avertissement: stackoverflow.com/a/47227886/4892874 Pour vérifier que vous êtes en utilisant le GPU: sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
  • Êtes-vous à l'aide de l'Anaconda ? si vous êtes, vous devez passer l'interprète de l'environnement particulier dans PyCharm.
  • Mappage de périphérique: pas de dispositifs connus. Je n'utilise pas d'Anaconda
  • Qu'est-ce que votre GPU?
  • NVIDIA 940mx, il est relatif à la nouvelle carte graphique Nvidia. @JorgeLeitão oui, alors je n'ai pas TensorFlow

InformationsquelleAutor Guruku | 2018-07-12