在 Ubuntu 上的 Surface Book 中检测 dGPU 驱动程序

在 Ubuntu 上的 Surface Book 中检测 dGPU 驱动程序

我有一台 W10 Surface Book,它具有对 Ubuntu16.04 的新原生支持。我正在尝试让 Tensorflow 工作,我已经正确安装了 cuda 和 cudnn,它们似乎都在工作。Tensorflow/python 也已正确设置。我现在遇到的问题是让 Ubuntu 在性能基础上识别 NVIDIA dGPU。

根据我的设备管理器,W10 正在检测它,但当我nvidia-sim在 Ubuntu 中尝试时,它说它无法与驱动程序通信。我也试过了echo $CUDA_VISIBLE_DEVICE,但没有。当我使用dpkg -l | grep nvidia以下方法检查驱动程序时,我得到了以下结果:

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

我认为可能是路径问题,但我已经按照安装说明中的建议设置了路径:

#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}}"

此时,我已经尝试重新安装/清除所有内容多次,但就是无法让它正常工作。

有人知道我能做什么吗?

相关内容