在Ubuntu上使用gpu?

时间:2017-10-09 03:12:40

标签: tensorflow gpu nvidia

我安装了CUDA驱动程序,Cudnn和tensorflow-gpu。我可以运行“nvidia-smi”和“nvcc --version”。但是,我还是不能用gpu,我不知道为什么,谁能告诉我? 我使用了一个例子:https://www.tensorflow.org/tutorials/using_gpu 但我得到的是: enter image description here

1 个答案:

答案 0 :(得分:0)

您需要使用与nvidia驱动程序相同的版本cuda-8。 如果您使用nvidia-384.xx,则需要cuda_8.0.xx_384.xx