花费了一天的时间终于将faster-rcnn跑通了。把整个流程记录下来,分享给大家,也有利于自己下次配置提供方便。第一次写博客,语句表达不清楚的地方请多包含。好了,下面进入正题。
主要参考:
http://blog.csdn.net/zoro_lov3/article/details/60581174
http://blog.csdn.net/hu_oo/article/details/54918256
http://blog.csdn.net/u013832707/article/details/53501478
运行环境:ubuntu16.04 N卡:GTX970
一:安装nvidia驱动
1. 首先在这个网站上查看属于自己电脑的显卡驱动(http://www.nvidia.com/Download/index.aspx?lang=en-us),下载下来。
2. 禁止集成的nouveau(鹦鹉螺)驱动
Ubuntu系统集成的显卡驱动程序是nouveau,它是第三方为NVIDIA开发的开源驱动,我们需要先将其屏蔽才能安装NVIDIA官方驱动。 将驱动添加到黑名单blacklist.conf中,但是由于该文件的属性不允许修改。所以需要先修改文件属性。
查看属性:
$sudo ls -lh /etc/modprobe.d/blacklist.conf
修改属性:
$sudo chmod 666 /etc/modprobe.d/blacklist.conf
使用gedit编辑器打开:
$sudo gedit /etc/modprobe.d/blacklist.conf
在文件的最后添加如下内容:
1. blacklist vga16fb
2. blacklist nouveau
3. blacklist rivafb
4. blacklist rivatv
5. blacklist nvidiafb
3. 开始安装nvidia显卡驱动
关闭当前图形工作界面(不要害怕,只是一小会的事情)
$sudo service lightdm stop
使用下列命令进入控制台
Ctrl + Alt + F1
安装显卡驱动的两种方式
方式一:
$sudo sh NVIDIA-Linux-x86_64-xxx.run
方式二:
1. sudo add-apt-repository ppa:graphics-drivers/ppa
2. sudo apt-get update
3. sudo apt-get install nvidia-384 #这里要写自己适合的gpu驱动版本
4. sudo apt-get install mesa-common-dev
5. sudo apt-get install freeglut3-dev
安装结束后进入原来的图形界面
$sudo service lightdm start
4. 查看安装的驱动
$nvidia-smi
出现下面的显卡信息表示安装成功:
到此,nvidia显卡的驱动就安装好了。
二:安装 cuda8.0
cuda是NVIDIA的编程语言平台,想使用gpu就必须要使用cuda。下载地址:(https://developer.nvidia.com/cuda-80-ga2-download-archive)
下载完cuda8.0后,执行如下语句,运行runfile文件:
$sudo sh cuda_8.0.61_375.26_linux.run
执行后会有一系列提示让你确认,但是注意,有个让你选择是否安装nvidia375驱动时,一定要选择否,其余的都直接默认或者选择是即可
安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc的尾部
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}