Keras: CuDNN não está disponível?

votos
1

Estou executando o exemplo kaggle_otto_nn.py de Kerascom backend de theano.

No seguinte impressão, linha 5, algo diz:

CNMeM é ativado com tamanho inicial: 90,0% da memória, CuDNN não disponível

Eu estou querendo saber, isso CuDNN not availableimporta desde que o dispositivo GPU é detectável? Estou executando o meu programa na GPU corretamente? Ou ele realmente não está funcionando em GPU.

cliu@cliu-ubuntu:keras-examples$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32,lib.cnmem=0.9 python kaggle_otto_nn.py
Using Theano backend.
/usr/local/lib/python2.7/dist-packages/Theano-0.8.0rc1-py2.7.egg/theano/tensor/signal/downsample.py:6: UserWarning: downsample module has been moved to the theano.tensor.signal.pool module.
  downsample module has been moved to the theano.tensor.signal.pool module.)
Using gpu device 0: Quadro K610M (CNMeM is enabled with initial size: 90.0% of memory, CuDNN not available)
Loading data...
9 classes
93 dims
Building model...
Training model...
Train on 52596 samples, validate on 9282 samples
Epoch 1/20
52596/52596 [==============================] - 6s - loss: 0.9420 - val_loss: 0.6269
Epoch 2/20
52596/52596 [==============================] - 6s - loss: 0.6955 - val_loss: 0.5817
...
Epoch 20/20
52596/52596 [==============================] - 6s - loss: 0.4866 - val_loss: 0.4819
Generating submission...
144368/144368 [==============================] - 1s     
Wrote submission to file keras-otto.csv.
Publicado 18/03/2016 em 19:32
fonte usuário
Em outras línguas...                            


1 respostas

votos
1

cuDNN é uma biblioteca por NVidia o que aumenta o desempenho de redes neurais na GPU. Portanto, o seu programa ainda é executado em GPU, mas muito mais lento do que poderia ter sido se você tinha instalado cuDNN.

Respondeu 19/03/2016 em 07:38
fonte usuário

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