[PYTHON] [Est-ce explosif!?] Configuration pour utiliser la version GPU de Tensorflow sous OSX

** * Ajouté le 12 mai 2016 Cet article est complètement obsolète car il a été fusionné dans le référentiel officiel Tensorflow. ** ** -> Le GPU OSX est pris en charge par Tensorflow Eh bien, la méthode d'installation ne change pas beaucoup.

** * Parce que ce n'est pas officiel, nous ne pouvons pas assumer la responsabilité quoi qu'il arrive. ** **

Je me suis souvenu que mon Macbook Pro (Retina, 15 pouces, mi-2014) avait un NVIDIA GeForce GT 750M en plus de ʻIntel Iris Pro, alors j'ai dit," 3.0 sur Cuda. J'ai pensé, et quand je cherchais un moyen, il y avait des gens qui le faisaient déjà à l'étranger, alors je quitterai la procédure lorsque j'y aurai fait référence. Cependant, il revient à ver0.6.0 Puisque bazel est utilisé, ceux qui ont installépip, Virtualenv , docker` devront peut-être d'abord installer à partir de la source.

• La plupart des produits Apple récents sont équipés de cartes graphiques Intel ou AMD, veuillez donc vérifier à l'avance. Vous pouvez vérifier les spécifications de «À propos de ce Mac» en haut à gauche à «Rapport système».

Peut-être le genre que tu peux aller

installer

Source de référence: Fabrizio Milo @ How to compile tensorflow with CUDA support on OSX

Tout d'abord, installez Cuda. J'étais «brasserie».

$ brew upgrade
$ brew install coreutils
$ brew cask install cuda

Vérifiez la version(7.5.Devrait être 20)
$ brew cask info cuda 

Téléchargez la bibliothèque libCudnn depuis NVIDIA. (Enregistrement requis) https://developer.nvidia.com/cudnn. Voici la version que j'ai téléchargée: cudnn-7.0-osx-x64-v4.0-prod.tgz

Déplacez le contenu de celui téléchargé vers les emplacements correspondants de lib et ʻinclude dans / usr / local / cuda /`, respectivement.

Ajoutez le chemin à .bash_profile

$ vim ~/.bash_profile
export DYLD_LIBRARY_PATH=/usr/local/cuda/lib:$DYLD_LIBRARY_PATH

Apportez Pull Request # 644 pour activer le GPU sur OSX dans le référentiel de Tensorflow

$ cd tensorflow
$ git fetch origin pull/664/head:cuda_osx
$ git checkout cuda_osx

Réinstaller Tensorflow

$ TF_UNOFFICIAL_SETTING=1 ./configure                                                                                                     
                                                                                                                                          
WARNING: You are configuring unofficial settings in TensorFlow. Because some external libraries are not backward compatible, these setting
s are largely untested and unsupported.                                                                                                   
                                                                                                                                          
Please specify the location of python. [Default is /usr/local/bin/python]:                                                                
Do you wish to build TensorFlow with GPU support? [y/N] Y                                                                                 
GPU support will be enabled for TensorFlow                                                                                                
Please specify the Cuda SDK version you want to use. [Default is 7.0]: 7.5                                                                
Please specify the location where CUDA 7.5 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:       
                                                                                                  
Please specify the Cudnn version you want to use. [Default is 6.5]: 4                                                                     
Please specify the location where cuDNN 4 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:        
Please specify a list of comma-separated Cuda compute capabilities you want to build with.                                                
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.                                            
Please note that each additional compute capability significantly increases your build time and binary size.                              
[Default is: "3.5,5.2"]: 3.0                                                                                                              
Setting up Cuda include                                                                                                                   
Setting up Cuda lib                                                                                                                       
Setting up Cuda bin                                                                                                                       
Setting up Cuda nvvm                                                                                                                      
Configuration finished   

$ bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer

$ bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

$ bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

$ pip install /tmp/tensorflow_pkg/tensorflow-Dans mon tmp-whl

Test de confirmation.py

import tensorflow as tf

# Creates a graph.
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)

# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

# Runs the op.
print sess.run(c)

Si l'erreur indique "Raison: image non trouvée", il semble que la bibliothèque Cuda est introuvable, vérifiez donc le chemin.

$ export DYLD_LIBRARY_PATH=/usr/local/cuda/lib:$DYLD_LIBRARY_PATH

Vous devriez pouvoir y aller.

Essayons de mesurer la vitesse de traitement avec le CNN de "Hiranaga MAIST".

。。。

Cette.

J'ai préparé l'image avec l'idée de **, mais cela s'est accéléré. ** **

Environ 52 minutes avec le processeur

CPU-MAIST.py

i 19900, training accuracy 1 cross_entropy 0.205204
test accuracy 0.943847
elapsed_time:3312.28295398[sec]

GPU-MAIST.py

i 19900, training accuracy 1 cross_entropy 0.0745807
test accuracy 0.945042                              
elapsed_time:1274.27083302[sec]

Environ 21 minutes avec GPU.

Si diverses autres applications sont en cours d'exécution, la mémoire du GPU peut être assez petite dans la partie Mémoire libre: du journal qui apparaît pendant l'exécution. S'il est trop petit, une erreur se produira en raison d'une mémoire insuffisante. C'est un ordinateur portable, donc je ne peux pas m'en empêcher. Il récupérera dès que vous déposerez l'application ou que vous la redémarrerez, donc c'est approprié.