如何为TensorFlow安装CUDA驱动程序

时间:2016-08-19 01:44:20

标签: cuda driver tensorflow ubuntu-15.04

我尝试build TensorFlow from source并使用GPU支持运行它。要安装工具包,我使用runfile来安装我使用附加驱动程序工具的驱动程序,因为我没有按照CUDA documentationstop lightdm以及{{1}中的指定启动Ubuntu进入文本模式它也不起作用,它给了我(也是sudo):

start lightdm

到目前为止,我可以从TensorFlow存储库构建一个版本。但是,当我试图按照操作方法

中的指定运行示例时
Name com.ubuntu.Upstart does not exist
显然无法找到GPU:

bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu

我在带有GTX950M的Acer笔记本上使用干净的Ubuntu 15.04安装。

有人能告诉我如何正确安装驱动程序吗?

1 个答案:

答案 0 :(得分:1)

您可以运行deviceQuery(cuda安装附带)吗?你能看到lspci / lsmod / nvidia-smi中存在nvidia吗?

lsmod |grep nvidia 
dmesg | grep -i nvidia
lspci | grep -i nvidia
nvidia-smi

您可以重新加载nvidia模块并查找错误消息

modprobe -r nvidia
dmesg | tail
sudo dmesg | grep NVRM

相关问题https://github.com/tensorflow/tensorflow/issues/601