在 Surface Book 上的 Ubuntu 中偵測 dGPU 驅動程式

在 Surface Book 上的 Ubuntu 中偵測 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}}"

此時,我一整天都多次嘗試重新安裝/清除所有內容,但我就是無法讓它發揮作用。

有人對我能做什麼有什麼想法嗎?

相關內容