Eu tenho um W10 Surface Book que tem o novo suporte nativo (ish) para o Ubuntu16.04. Estou tentando fazer com que o Tensorflow funcione, instalei o cuda e o cudnn corretamente e ambos parecem estar funcionando. Tensorflow / python também são configurados corretamente. O problema que estou tendo agora é fazer com que o Ubuntu reconheça o dGPU da NVIDIA na base de desempenho.
De acordo com o meu gerenciador de dispositivos, o W10 está detectando, mas quando eu tento o nvidia-sim no Ubuntu ele diz que não pode se comunicar com os drivers. Eu também tento echo $CUDA_VISIBLE_DEVICE e não há nenhum. Quando eu verifico os drivers usando dpkg -l | grep nvidia , é isso que recebo:
ii nvidia-384 384.90-0ubuntu0.16.04.1 amd64 NVIDIA binary driver - version 384.90
ii nvidia-384-dev 384.90-0ubuntu0.16.04.1 amd64 NVIDIA binary Xorg driver development files
ii nvidia-modprobe 384.81-0ubuntu1 amd64 Load the NVIDIA kernel driver and create device files
ii nvidia-opencl-icd-384 384.90-0ubuntu0.16.04.1 amd64 NVIDIA OpenCL ICD
ii nvidia-prime 0.8.2 amd64 Tools to enable NVIDIA's Prime
ii nvidia-settings 384.90-0ubuntu0~gpu16.04.1 amd64 Tool for configuring the NVIDIA graphics driver
Achei que talvez fosse um problema de caminho, mas defini meus caminhos conforme sugerido nas instruções de instalação:
#cuda
export PATH="/usr/local/cuda-8.0/bin${PATH:+:${PATH}}"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"
Neste ponto, tentei reinstalar / limpar tudo várias vezes durante o dia todo, mas não consigo fazer isso funcionar.
Alguém tem alguma ideia do que eu posso fazer?