2015-10-12 1 views
3

Manquant d'autorisations root J'ai installé Theano et Lasagne dans un virtualenv python3 sous Ubuntu 14.04.Utiliser CUDA dans virtualenv pour servir Theano

En cours d'exécution de code, j'obtiens un ImportError: dnn not available, qui, pour autant que je vois, résulte de l'en-tête CUDA manquant, bien qu'il soit installé à /usr/local/cuda-6.5/targets/x86_64-linux/include/cudnn.h.

Après la theano documentation je mets la variable d'environnement CUDA_ROOT=/usr/local/cuda-6.5, ce qui donne une erreur supplémentaire:

ERROR (theano.sandbox.cuda): Failed to compile cuda_ndarray.cu: libcublas.so.6.5: cannot open shared object file: No such file or directory

Est-il possible d'inclure cette bibliothèque CUDA dans le virtualenv?

+0

J'ai travaillé autour de l'erreur theano.sandbox suivante [http://stackoverflow.com/questions/20026127/oserror-libcublas-so-3-cannot-open- fichier-objet-partagé-non-tel-fichier-ou-directo) instructions. Mais toujours l'erreur d'importation dnn. – frthjf

+0

La [documentation theano] (http://deeplearning.net/software/theano/library/sandbox/cuda/dnn.html) suggère que vous définissiez également votre CPATH? – inJeans

+0

Merci pour l'indice, mais aucun résultat jusqu'à présent. Je vais essayer de déboguer la fonction dnn_available() pour en savoir plus. – frthjf

Répondre

3

Enfin trouvé la réponse. En plus des variables d'environnement j'ai oublié de configurer le .theanorc dans mon répertoire personnel comme le theano documentation suggéré:

$ cat /home/user/.theanorc 
[global] 
floatX = float32 
device = gpu0 

[cuda] 
root = /usr/local/cuda-6.5 

Merci quand même! J'ai appris la leçon à lire docs attentivement ...