Caffe Ubuntu14.04 + CUDA 8 (支持GTX1080 1070等Pascal架构显卡)

1. 前言

本教程使用的系统是Ubuntu 14.04 LTS 64-bit,使用的CUDA版本为8。

理论上本教程支持Pascal架构显卡,如游戏卡GeForce GTX1070,GTX 1080,新Titan X;以及刚刚推出的计算卡Tesla P100。

如果你在安装时使用计算卡进行GPU加速,而用于显示的显卡不是NVIDIA显卡,则可能导致因为安装NVIDIA驱动时自动加载X Server配置而无法启动图形界面。这时可以使用服务器版的Ubuntu,或者换用CentOS等。

2. 安装基础依赖项

sudo apt-get install build-essential
sudo apt-get install cmake git

3. 安装CUDA

首先需要关闭图形界面。

使用Ctrl+Alt+F1进入虚拟控制台,然后输入用户名和密码登陆。然后关闭lightdm:

sudo /etc/init.d/lightdm stop

CUDA 8需要注册Accelerated Computing Developer Program,然后可以免费下载。

在如下地址进行注册和下载:

点击这里

下载完成后,前往指定的目录,然后执行:

sudo dpkg -i cuda-repo-ubuntu1404-8-0-rc_8.0.27-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

安装完成后,重启电脑:

sudo reboot

之后打开Terminal,添加环境变量。使用gedit打开如下文档:

sudo gedit /etc/profile

在文件末尾添加:

PATH=/usr/local/cuda/bin:$PATH
export PATH

保存完成后,执行如下命令使环境变量立即生效:

source /etc/profile

然后还需要添加lib的路径:

sudo gedit /etc/ld.so.conf.d/cuda.conf

在文件中写入如下内容然后保存:

/usr/local/cuda/lib64

之后执行如下命令使之生效:

sudo ldconfig

执行完了这些操作之后,还可以安装CUDA SAMPLES来检测CUDA是否运行正常,鉴于这不是CUDA编程教程,本教程暂不介绍。

4. 安装其他的一些依赖项

sudo apt-get install freeglut3-dev libx11-dev libxmu-dev libxi-dev libglu1-mesa-dev
sudo apt-get install libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

PS:复制粘贴太长的命令可以能因为浏览器的原因导致输入了多余的换行符,如果复制粘贴带有换行符的命令进Terminal,会被当做两条命令来执行,一定要注意这一点。

5. 安装ATLAS

本步骤可以用OpenBLAS或者Intel MKL替代。我在E5-2690v2+GTX780的平台上测试过这三种库,性能相差无几,这里就介绍其中一种:

sudo apt-get install libatlas-base-dev

然后自动安装完成即可。

目前intel MKL是收费软件,OpenBLAS可以免费下载和安装。如果使用这两种库,编译Caffe时要在Makefile.config做出对应的修改。

6.下载Caffe

从GitHub上直接下载Caffe的最新版

git clone https://github.com/BVLC/caffe.git

也可以从Caffe的GitHub工程中下载Caffe的历史版本:

https://github.com/BVLC/caffe

7. 安装Python

先安装相关依赖项

sudo apt-get install python-dev python-pip

转到下载的caffe的目录下,然后转到python目录下

cd python

执行如下命令:

for req in $(cat requirements.txt); do sudo pip install $req; done

等待其自动安装即可。

8. 编译Caffe

在caffe目录下,执行:

cp Makefile.config.example Makefile.config

然后打开Makefile.config,根据自己的需要修改相关参数。

进行编译:

make all -j10
make test
make runtest

PS:-j10指10线程同时编译,根据自己CPU的特点调整该参数,如果不了解自己CPU,直接执行make即可。

然后可以根据需求编译pycaffe:

make matcaffe
make pycaffe

然后caffe就安装结束了。

9. 运行minist demo

转到 Caffe 目录下

sh data/mnist/get_mnist.sh
sh examples/mnist/create_mnist.sh
sh examples/mnist/train_lenet.sh

如果运行正常,caffe就可以正常工作了。

如需配置MATLAB和cuDNN,可以参考这篇教程:点击这里

本教程编写参考了如下教程,特此鸣谢:http://www.cnblogs.com/platero/p/3993877.html

Caffe的官方网站是 http://caffe.berkeleyvision.org/

如果有任何问题可以联系我:[email protected]

最近更新于2016-09-09 13:33:06

时间: 2024-12-28 16:40:01

Caffe Ubuntu14.04 + CUDA 8 (支持GTX1080 1070等Pascal架构显卡)的相关文章

Deep Learning 学习总结(一)—— Caffe Ubuntu14.04 CUDA 6.5 配置

Caffe (Convolution Architecture For Feature Extraction)作为深度学习CNN一个非常火的框架,对于初学者来说,搭建Linux下的Caffe平台是学习深度学习关键的一步,其过程也比较繁琐,回想起当初折腾的那几天,遂总结一下Ubuntu14.04的配置过程,方便以后新手能在此少走弯路. 1. 安装build-essentials 安装开发所需要的一些基本包 sudo apt-get install build-essential 2. 安装NVID

ubuntu14.04+cuda8.0(GTX1080)+caffe安装

1. ubuntu14.04安装 分区:usr/local 300G boot 200M / 根目录 300G 交换空间 20G (内存16G) 其余都是/home 重装之后,下载NVIDIA-Linux-x86_64-367.27.run,下载地址为:http://www.geforce.cn/drivers/results/104314 将其与cuda_8.0.27_linux.run(下载地址为:https://developer.nvidia.com/cuda-toolkit 不过需要注

待续Oracle VM VirtualBox+ubuntu14.04+cuda+caffe

Oracle VM VirtualBox下载 ubuntu14.04 先安装virtualbox,再在上面装ubuntu14.04.注意要安装增强功能(启动虚拟机后,选择“Devices”菜单->选择“Insert Guest Additions CD Images”选项.如果没看见devices,按右crtl+c),否则屏幕显示不全. caffe的安装(暂时还没有安好,遇到的问题:安装完cuda之后重启,登录界面循环),参考的安装教程如下: 多版本安装教程 (注意要禁用Nouveau,安装cu

Caffe+Ubuntu14.04+CUDA7.5 环境搭建(新人向)指南

序 本文针对想学习使用caffe框架的纯新手,如果文中有错误欢迎大家指出. 由于我在搭建这个环境的时候参考了许多网上的教程,但是没有截图,所以文中图片大多来源于网络. 本文没有安装matlab的步骤,因此需要安装并配置matlab的同学请百度matlab安装. 1. 在Win10环境下搭建Ubuntu14.04双系统 操作事先请准备好: 一个空的U盘,最好大于4G. 去Ubuntu官网下载一个Ubuntu14.04的iso镜像文件,注意要64位系统. 下载一些用的到的小工具,如EsayBCD,u

Caffe+Ubuntu14.04+CUDA7.5安装笔记

ubuntu 14.04安装 先到官网下载ubuntu14.04,网址:http://www.ubuntu.com/download/desktop ubuntu-14.04.4-desktop-amd64.iso 参考:Ubuntu14.04 安装及使用:[1]制作安装U盘 制作安装U盘 然后参考:Ubantu14.04安装教程 安装系统 分区: boot 设置 200M 主分区 / 设置 50000M swap 设置 4000M home 设置 剩余M cuda7.5安装 cuda7.5下载

Ubuntu14.04配置Apache支持多个站点

怎样在一个Ubuntu的机器上(虚拟机)配置Apache支持多个网站呢? 比如你有一台独立的Ubuntu虚拟机,配有一个外网的IP(45.46.47.48),并且注册了两个域名AAA.com和BBB.com,将这两个域名DNS解析到你虚机的IP地址.假设你已经安装好了Apache,一切都是默认的设置. 我们需要在这一个server上面,同时host AAA.com,BBB.com 第一步:修改hosts文件 在Ubuntu系统中,hosts文件目录为/etc/hosts,可以用vi编辑 sudo

caffe+Ubuntu14.0.4 64bit 环境配置说明(无CUDA,caffe在CPU下运行) --for --Amd

caffe是一个简洁高效的深度学习框架,具体介绍可以看这里,caffe环境配置过程可以参考这里,我在搭建环境时搜集了许多资料,这里整理了一下,介绍一下caffe在无CUDA的环境下如何配置. 1. 安装build-essentials 安装开发所需要的一些基本包 sudo apt-get install build-essential 如果出现essential包不可用的情况,可以执行下列命令解决: sudo apt-get update 2. 安装ATLAS for Ubuntu 执行命令:

ubuntu14.04安装及web环境的搭建

在进行接下来的工作之前首先得准备一个Ubuntu的启动U盘以及腾出一个至少50G的系统盘(在网上看到有人用20G的,不知道效果怎么样).关于Ubuntu启动U盘大家可以参考这个帖子 http://tieba.baidu.com/p/2931186318. 需要准备的东西ok之后,就可以开始装机了,首先设置电脑的第一启动项为你的U盘,进入到Ubuntu系统之后,按照提示一步一步next(关于连不联网的问题,我觉得最好还是不要联网,因为联网后会给你下一些其他的东西,导致装系统的速度变慢,具体什么东西

ubuntu14.04下安装lec10.1过程记录

0.lec为Candence公司推出的形式验证工具,由于要学习IC后端的知识,打算在ubuntu14.04下安装一套后端工具,建立后端环境,通过练习完成后端知识的入门,安装过程记录下来,留作以后查看,如果对您有一些帮助,倍感欣慰! 1.安装源文件 >lec10.1:http://pan.baidu.com/s/1o6wqu5G 3.安装与破解 >解压lec_10.1.tgz到安装目录 >设置环境如下: #add for lec export LEC_HOME=/usr/cad/lec_1