search

drethage / speech-denoising-wavenet

A neural network for end-to-end speech denoising
MIT License
673 stars 165 forks source link

Nvidia driver version exception #25

Open gutsy-robot opened 5 years ago

gutsy-robot commented 5 years ago

I am trying to run this model on my files. I have followed the instructions in the documentation. When I run this command:

THEANO_FLAGS=optimizer=fast_compile,device=gpu KERAS_BACKEND=theano python main.py --mode inference --target_field_length 16001 --batch_size 4 --config sessions/001/config.json --noisy_input_path ../noisy

I get an exception,

Exception: The nvidia driver version installed with this OS does not give good results for reduction.

I am cuda9.0 and I had installed all the requirements in the requirements.txt. Has anyone faced a similar issue.

jmqpool commented 5 years ago

I experienced the same issue and was able to resolve it by using cuda-8.0 with cudnn-5.1.