声明:本次博客主要记录自己配置环境的过程以及中间遇到的问题和解决问题的方案,如果能有幸帮助到他人不胜荣幸,如果错误,望指出,以便及时更改。另外本方法是博主在自己的机子上一步一步进行的,确保可行,但是具体到不同的机子可能会有不同的问题,仅供参考。
环境:纯净的ubuntu16.04 显卡: GTX1050
安装流程:
1、安装英伟达显卡驱动
2、安装cuda8.0
3、安装cudnn
4、安装anaconda2
5、配置caffe环境,配置python接口
一:安装英伟达显卡驱动
本人的显卡为GTX1050,在网上看了诸多教程,好多教程都说要提前去官网下载好驱动,然后关闭图形界面,卸载ubuntu自带的驱动之类的,这种方法我亲身实验过几次,效果不好,装不上,总出问题,最明显的就是在登录界面循环,最后不得不把驱动卸载才能进去。本次方法使用ppa源直接安装,如下:
首先先去官网看一下自己GPU对应的驱动的版本号,这个不需要下载。
我使用的版本是375。
1、电脑连上网
2、按下ctrl+alt+F1进入tty模式(黑窗口)
3、输入账户名和密码(建议此处不用小键盘)
4、依次运行下列语句
sudo apt-get purge nvidia-* //删除现有驱动
sudo add-apt-repository ppa:graphics-drivers/ppa //添加源
sudo apt-get update //更新源
sudo apt-get install nvidia-375 //nvidia-后面的数字是版本号
sudo reboot //重启
电脑重启之后,打开终端,输入:
nvidia-smi
出现结果如下图:
或者直接右上角,关于这台计算机,如下
OVER !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
二、安装cuda8.0
2.1 gcc版本问题
cuda8.0在ubuntu16.04中好像支持的gcc版本不超过5.3,提前先查一下自己的gcc版本,如果不符合要求就要降级。
查询方式:
gcc --version
g++ --version
博主的版本如下:
不需要降级。
如果需要降级的话推荐下面这种方式:
(1)下载gcc/g++ 4.7.x
sudo apt-get install -y gcc-4.7
sudo apt-get install -y g++-4.7
(2)链接gcc/g++实现降级
cd /usr/bin
sudo rm gcc
sudo ln -s gcc-4.7 gcc
sudo rm g++
sudo ln -s g++-4.7 g++
2.2 安装cuda8.0
请提前下载好cuda 8.0的runfile安装包,请自行百度,我用的是.run格式,安装包放在下载目录下。
1、cd到下载目录
2、终端输入下列代码
sudo sh cuda_8.0.61_375.26_linux.run --override