pythontensorflowcudascreencuda-driver

ImportError: libcuda.so.1: cannot open shared object file

When I run my code with TensorFlow directly, everything is normal.

However, when I run it in a screen window, I get the following error.

ImportError: libcuda.so.1: cannot open shared object file: No such file or directory

I have tried the command:

source /etc/profile

But it doesn't work.

Cause I use ssh to connect to the servers, the screen is necessary.

How can I fix it?

Solution

  • Try to put libcuda.so.1 path to LD_LIBRARY_PATH environment variable.

    example:

    export LD_LIBRARY_PATH=/path/of/libcuda.so.1:$LD_LIBRARY_PATH