最近也纠结在这个问题上几天,每次装好cuda之后,运行theano测试程序时,总是显示CUDA is installed, but device gpu is not available (error: cuda unavailable)。
但cuda的测试程序是可以跑通的,一度怀疑是theano有bug,最后终于找到了解决方法。
首先我们需要安装nvidia的驱动,使用命令
ubuntu-drivers devices
查看我们的显卡所支持的驱动,然后
sudo apt-get install nvidia-*
安装对应版本的驱动。重启!