Como posso usar o meu GPU em ipython Notebook?

votos
6

SO: Ubuntu 14.04LTS
Idioma: Python Anaconda 2,7 (keras, Théano)
GPU: GTX980Ti CUDA: 7,5 CUDA

Eu quero correr código python keras em IPython Notebook usando minha GPU (GTX980Ti)
Mas eu não posso encontrá-lo.

Eu quero testar o código abaixo. Quando eu executá-lo em ao terminal Ubuntu, eu comando como abaixo (Ele usa GPU bem. Ele não tem qualquer problema)

Primeiro eu definir o caminho como abaixo

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH 

Segundo eu executar o código como abaixo

THEANO_FLAGS='floatX=float32,device=gpu0,nvcc.fastmath=True'  python myscript.py


E funciona bem.

Mas quando eu executar o código em PyCharm (IDE Python) ou quando eu executá-lo em ipython Notebook, Ele não usa GPU. Ele só usa CPU

myscript.py código é como a seguir.

from theano import function, config, shared, sandbox
import theano.tensor as T
import numpy
import time
vlen = 10 * 30 * 768  # 10 x #cores x # threads per core
iters = 1000
rng = numpy.random.RandomState(22)
x = shared(numpy.asarray(rng.rand(vlen), config.floatX))
f = function([], T.exp(x))
print(f.maker.fgraph.toposort())
t0 = time.time()
for i in xrange(iters):
    r = f()
t1 = time.time()
print(Looping %d times took %f seconds % (iters, t1 - t0))
print(Result is %s % (r,))
if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()]):
    print('Used the cpu')
else:
    print('Used the gpu')

Para resolvê-lo, eu forçar a GPU uso código como abaixo (Inserir duas linhas mais em myscript.py)

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

Em seguida, ele gera o erro como abaixo

ERROR (theano.sandbox.cuda): nvcc compiler not found on $PATH. Check your nvcc installation and try again.

como fazer isso??? Passei dois dias ..
E eu certamente fiz a maneira de usar o arquivo '.theanorc' no diretório home.

Publicado 17/12/2015 em 04:59
fonte usuário
Em outras línguas...                            


1 respostas

votos
2

Estou usando Theano em um ipython notebook fazendo uso da GPU do meu sistema. Esta configuração parece funcionar bem no meu sistema. (Macbook Pro com GTX 750M)

Meu arquivo ~ / .theanorc:

[global]
cnmem = True
floatX = float32
device = gpu0

Várias variáveis ​​de ambiente (eu uso um ambiente virtual (macvnev):

echo $LD_LIBRARY_PATH
/opt/local/lib:

echo $PATH
/Developer/NVIDIA/CUDA-7.5/bin:/opt/local/bin:/opt/local/sbin:/Developer/NVIDIA/CUDA-7.0/bin:/Users/Ramana/projects/macvnev/bin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin

echo $DYLD_LIBRARY_PATH
/Developer/NVIDIA/CUDA-7.5/lib:/Developer/NVIDIA/CUDA-7.0/lib:

Como eu corro ipython notebook (Para mim, o dispositivo é gpu0):

$THEANO_FLAGS=mode=FAST_RUN,device=gpu0,floatX=float32 ipython notebook

Saída de $nvcc -V:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2015 NVIDIA Corporation
Built on Thu_Sep_24_00:26:39_CDT_2015
Cuda compilation tools, release 7.5, V7.5.19

A partir do seu post, provavelmente você definir o errado variável $ PATH.

Respondeu 10/02/2016 em 10:19
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more