Tensorflow com CUDA 8.0 rc

Question

Quando tento importar o tensorflow no Python, recebo esse erro.

ImportError: libcudart.so.7.5: cannot open shared object file: No such file or directory

Estou usando o CUDA 8.0 RC e o Ubuntu 16.04.

    

Answer 1

Para o CUDA 8.0 no Ubuntu 16.04 você precisa construir e instalar o TF a partir da fonte, durante a fase de configuração ele solicitará a versão CUDA, especifique 8.0

    

Answer 2

Como diz em TensorFlow docs, você deve instalar o CUDA via pip, e então o pacote tensorflow como ele diz. Então há uma ativação, antes de usar.

$ source ~/tensorflow/bin/activate  # If using bash.
$ source ~/tensorflow/bin/activate.csh  # If using csh.
(tensorflow)$  # Your prompt should change.
# Run Python programs that use TensorFlow.
...
# When you are done using TensorFlow, deactivate the environment.
(tensorflow)$ deactivate

Espero que isso ajude.

    

Answer 3

Outra opção é apenas remover o cuda 8.0 e instalar o cuda 7.5, o que eu fiz. Remover cuda 8.0 -

sudo dpkg -r cuda

Baixe o cuda 7.5 do site nvidia. Então, faça -

sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
sudo apt-get update

Depois, acesse o gerenciador de pacotes sinápticos e procure por cuda . Selecione cuda-7-5 e instale tudo associado. Você tem que fazer isso a partir do gerenciador de pacotes sinápticos. Se você fizer sudo apt-get install do terminal, ele irá instalar novamente o cuda 8.0.

Espero que isso evite que alguém tire o cabelo da frustração.

    

Answer 4

Você diz que tem CUDA 8.0, mas seu libcudart parece ser a versão 7.5. Pode ser necessário remover o CUDA e começar de novo, rm -rf /usr/local/cuda-<7.5 or 8.0> Lembre-se de substituir o < > seção com sua (s) versão (ões) real (is) e execute-a como sudo . Depois de fazer uma nova instalação, crie um arquivo .conf em /etc/ld.so.conf.d /:

sudo nano /etc/ld.so.conf.d/nvidia.conf

Adicione isto:

/usr/local/cuda-8.0/lib64
/usr/local/cuda-8.0/lib  # you probably don't need this line; check
                         # if the directory exists

Em seguida, execute sudo ldconfig . Espero que isso te traga alegria.