本人经过数次个人摸索,总算装上了caffe,踩了不少坑,写这篇博客希望可以让别人少走弯路
一.nvidia 驱动安装
因为我是重装了系统所以之前没有安过nvidia驱动,如果之前有装得话,可以自行删除,先
通过快捷键Ctrl+Alt+T打开终端
sudo apt-get remove --purge nvidia-*
下面开始安装
首先添加官方源
sudo add-apt-repository ppa:graphics-drivers/ppa
然后刷新软件库并安装(首先先去http://www.nvidia.com/Download/index.aspx?lang=en-us查询自己适合的驱动)
sudo apt-get update
sudo apt-get install nvidia-367 nvidia-settings nvidia-prime
上面的nvidia-367根据你查询的结果自行更改
Ps:这方法适用于Ubuntu16.04其他版本可能有问题
安装完成之后重启电脑
然后在命令行输入
nvidia-settings
应该会出现如下界面
这就说明成功了,至此笔者是没遇到过什么问题,如果有问题的话自行google。
二
CUDA8.0安装与配置(先要随便注册个帐号)
1.首先先去官网下载
Base Installer和Patch 1都要下载
下载完后进行md5检测,如果不对的话就重新下载吧。
下载完成后
cd 进入你下载的目录
chmod 777 cuda_8.0.27_linux.run //根据自己下载的文件自行修改后面的版本号
sudo ./cuda_8.0.27_linux.run --override //不加--override会出现不支持编译器的问题
然后根据他的提示安装,注意问你是否要安装nvidia驱动的时候输入n,千万不要安装,其他的一路默认就行了
安装完成后会出现
============ Summary ============
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Installed in /home/tangdh/NVIDIA_CUDA-8.0_Samples
类似的界面,那个(tangdh根据每个人的不同而不同)
然后类似的方法安装你下载的另一个文件,先cd进入下载目录
chmod 777 cuda_8.0.27.1_linux.run
sudo ./cuda_8.0.27.1_linux.run
如果有说编译器不支持就输入
sudo ./cuda_8.0.27.1_linux.run --override
根据提示一路下去
然后就安装好了。
下面我们要配置环境
sudo g