search

TobyPDE / FRRN

Full Resolution Residual Networks for Semantic Image Segmentation
MIT License
278 stars 93 forks source link

ERROR Cannot create cuDNN spatial log softmax. #33

Open jhtao1860 opened 7 years ago

jhtao1860 commented 7 years ago

2 warning and 1 error Using gpu device 0: GeForce GTX 1080 Ti (CNMeM is disabled, cuDNN 6021) /usr/local/lib/python2.7/dist-packages/theano/sandbox/cuda/init.py:600: UserWarning: Your cuDNN version is more recent than the one Theano officially supports. If you see any problems, try updating Theano or downgrading cuDNN to version 5. warnings.warn(warn)

/usr/local/lib/python2.7/dist-packages/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.")

2017-09-08 19:34:11,781 ERROR Cannot create cuDNN spatial log softmax. Install cuDNN and make sure that theano uses the GPU.

need your help

daicoolb commented 6 years ago

@jhtao1860
First , you can touch a .theanorc file in your home location, Then add " [lib]\n cnmem=1" to it. Finally ,you should update your theano version to an appropriate version.