error if using deepspeech-gpu

MainRo / docker-deepspeech-server

A dockerfile to run deepspeech-server

Mozilla Public License 2.0

30 stars 14 forks source link

error if using deepspeech-gpu #2

Open kenyeung128 opened 6 years ago

kenyeung128 commented 6 years ago

hi, if i tried to use deepspeech-gpu instead, it throws the error as below

ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

I am using nvidia-docker

MainRo commented 6 years ago

The cublas library is part of the cuda drivers installation. You should find it in /usr/local/cuda-8.0/lib64/. I did not try nvidia-docker yet. Maybe we need to give it the cuda path and library path, just like for direct usage: PATH=$PATH:/usr/local/cuda-8.0/bin LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64