Comment puis-je exécuter théano sur GPU

Si j'exécute le code suivant avec python 3.5

import numpy as np
import time
import theano
A = np.random.rand(1000,10000).astype(theano.config.floatX)
B = np.random.rand(10000,1000).astype(theano.config.floatX)
np_start = time.time()
AB = A.dot(B)
np_end = time.time()
X,Y = theano.tensor.matrices('XY')
mf = theano.function([X,Y],X.dot(Y))
t_start = time.time()
tAB = mf(A,B)
t_end = time.time()
print ("NP time: %f[s], theano time: %f[s] **(times should be close when run
on CPU!)**" %(np_end-np_start, t_end-t_start))
print ("Result difference: %f" % (np.abs(AB-tAB).max(), ))

Je obtenir la sortie

NP time: 0.161123[s], theano time: 0.167119[s] (times should be close when
run on CPU!)
Result difference: 0.000000

il dit que si les temps sont proches, cela signifie que je suis en cours d'exécution sur mon CPU.

Comment puis-je exécuter ce code sur mon GPU?

REMARQUE:

  • J'ai une station de travail avec la technologie Nvidia Quadro k4200.
  • J'ai installé Cuda toolkit
  • J'ai travaillé avec succès un cuda vectorAdd exemple de projet sur VS2012.

OriginalL'auteur babeyh | 2015-11-04

  1. 12

    Vous configurez Théano d'utiliser un GPU en spécifiant le device=gpu dans Théano de la config. Il existe deux principales méthodes pour le réglage de la config: (1) dans le THEANO_FLAGS variable d'environnement, ou (2) par l'intermédiaire de l' .theanorc fichier. Les deux méthodes, et tous Théano de configuration de drapeaux,documenté.

    Vous savez que Théano est l'utilisation du GPU si, après l'appel de import theano vous voyez un message qui ressemble à quelque chose comme ceci

    Using gpu device 0: GeForce GT 640 (CNMeM is disabled)

    Les détails peuvent varier pour vous, mais si aucun message n'apparaît à tous, alors Théano est à l'aide de la CPU.

    Note également que, même si vous voyez le GPU message, votre calcul graphique peut ne pas s'exécuter sur le processeur graphique. Pour voir quelles parties de votre calcul sont en cours d'exécution sur le processeur graphique imprimer son compilé et optimisé graphique

    f = theano.function(...)
theano.printing.debugprint(f)

    Opérations qui commencent par le préfixe " Gpu " sera exécuté sur le processeur graphique. Des opérations qui n'ont pas de préfixe de leur nom sera exécuté sur le PROCESSEUR.

    merci. mais dans mes variables d'environnement, il n'y a pas de variable sur théano. le document de dire un fichier d'appel .theanorc, dans mon répertoire home, le fichier est absent. comment j'ai mis "périphérique" de la valeur ?
    Si THEANO_FLAGS n'existe pas, créez-le!
    import theano device='gpu0' vérifier la valeur de périphérique print(theano.config.device) il est " cpu " de nouveau.
    comment je créer ce drapeaux ?
    Dépend de votre système d'exploitation. Pour Windows, essayez microsoft.com/resources/documentation/windows/xp/all/proddocs/.... Pour Linux, essayez de digitalocean.com/community/tutorials/...

    OriginalL'auteur Daniel Renshaw

  2. 7

    Si vous êtes sur Linux, créer une .theanorc fichier dans votre dossier d'accueil et d'ajouter de la façon suivante pour définir théano à exécution sur GPU.

    [global]
device = gpu
floatx = float32
    juste un détail. c'est float32, pas floar32 (faute de frappe). Je ne peux pas le modifier 🙁
    pour la version 1.0.4; device=cuda

    OriginalL'auteur Dherath

  3. 5

    Alternativement, si vous souhaitez utiliser le GPU programattically:

    import theano.sandbox.cuda
theano.sandbox.cuda.use("gpu0")

    Vous devriez voir un message comme ceci:

    Using gpu device 0: Tesla K80

    Utile si l'environnement que vous utilisez n'est pas facile à configurer.

    Cela fonctionne pour théano v0.9. et il est maintenant obsolète. You are importing theano.sandbox.cuda. This is the old GPU back-end and is removed from Theano. Use Theano 0.9 to use it. Even better, transition to the new GPU back-end! See https://github.com/Theano/Theano/wiki/Converting-to-the-new-gpu-back-end%28gpuarray%29

    OriginalL'auteur Erik Aronesty