在Ubuntu上使用gpu？

Question

我安裝了CUDA驅動程序，Cudnn和tensorflow-gpu。我可以運行「nvidia-smi」和「nvcc --version」。但是，我仍然不能使用GPU，我不知道爲什麼，任何人都可以告訴我？ 我用一個例子從這個：https://www.tensorflow.org/tutorials/using_gpu 但我得到的是： enter image description here

Answer 1

您需要使用同一版本的CUDA-8作爲NVIDIA的驅動程序。 如果您使用nvidia-384.xx，則需要cuda_8.0.xx_384.xx