主要参考网址:
caffe官网:http://caffe.berkeleyvision.org/install_apt.html
http://www.linuxdiyf.com/linux/12613.html
按着流程走,最后编译成功。
如果出现nvidia-smi has failed because it couldn't communicate with the nvidia driver. make sure that the latest nvidia driver is installed and running.这种情况一般是内核升级导致的(内核升级一般是优化之前的bug)
解决办法三种:
1、重装cuda
2、修改启动的内核版本,之前安装是哪个版本就启动哪个版本
3、安装DKMS
DKMS全称是DynamicKernel ModuleSupport,它可以帮我们维护内核外的驱动程序,在内核版本变动之后可以自动重新生成新的模块。在使用dkms之前首先需要确保系统中已经安装了DKMS,在Ubuntu下可以执行下面这个命令安装。
sudo apt-get install dkms
通过cd /usr/src,然后ls一下,查看你对应安装的驱动版本,找到nvidia对应的驱动版本,重新生成对应的驱动模块,如果驱动是nvidia-440.31
sudo dkms install -m nvidia -v 440.31
更多具体信息可以参考:https://blog.csdn.net/Felaim/article/details/100516282