Ubuntu 16.04 安装CUDA8.0+Cudnn6.0+TensorFlow+Caffe安装

参考博客:
Ubuntu16.04 Caffe 安装步骤记录(超详尽):http://blog.csdn.net/yhaolpz/article/details/71375762
ubuntu16.04安装caffe:http://blog.csdn.net/u011272513/article/details/52103453#

环境说明:

1)Ubuntu 16.04:ubuntu-16.04.3-desktop-amd64.iso
2)采用CUDA 8.0 :cuda_8.0.44_linux.run
3)Cudnn 6.0:cudnn-8.0-linux-x64-v6.0.tgz
4)TensorFlow
5)Caffe
6)Python 2.7
7)pip 9.0.1

具体安装步骤:

一:CUDA 8.0安装

(包括NVIDIA显卡驱动和CUDA库文件)
通过英伟达安装包来分别安装库的驱动程序而不是使用apt-get 来安装。如果只有CPU,则可以跳过这步。
1)打开终端,加入官方PPa源:

sudo  add-apt-repository ppa:graphics-drivers/ppa
 
 
  • 1

2)按enter键确认,之后更新软件库并安装最新驱动

    sudo apt-get update
    sudo apt-get install nvidia-367 nvidia-settings nvidia-prime
 
 
  • 1
  • 2

Note:该Ubuntu 镜像文件在sudo apt-get update 时报错,解决方法:
输入命令:sudo apt-get purge libappstream3
之后再次执行 sudo apt-get update命令就正常了。

3)安装完成后,通过命令查看是否安装成功

     nvidia-settings
 
 
  • 1

此时,安装完成后,要重启电脑,有如下效果则安装完成,否则就说明安装有问题,尝试关闭UEFI保护试试。这里写图片描述

4)从官网上下载CUDA 8.0文件,下载cuda_8.0.44_linux.run文件,将其放至~/CUDA文件夹下,cd命令进入该文件所在目录,进行MD5检验,确保安装包正常。

  sudo md5sum cuda_8.0.44_linux.run
 
 
  • 1

5)检验正确后,输入如下代码安装(第一行修改文件权限,第二行执行文件安装):

chmod 777 cuda_8.0.44_linux.run
sudo ./cuda_8.0.44_linux.run  --override
 
 
  • 1
  • 2

这里写图片描述

执行代码后,跳出相关信息,左下方出现–更多–(0%) 这样,
这里写图片描述
一直按Enter键直到跳到100%结束退出,接下来要求输入相关配置命令,如下图,注意别弄错了,第二行配置询问是否要安装NVIDIA图像驱动器时,一定要选n,否则系统会报错,具体如下图:
这里写图片描述
安装完成后出现如下说明安装正确:
这里写图片描述

二、Cudnn 6.0安装
注意下载的版本为:cudnn-8.0-linux-x64-v6.0.tgz
其中CUDA对应的5.1版本,在后面安装TensorFlow时会出现问题。
1)进入文件所在目录进行解压:

tar zxvf cudnn-8.0-linux-x64-v6.0.tgz
 
 
  • 1

2)解压后有个cuda文件,内有include 和 lib64两个文件,进入include文件夹,执行如下命令:

sudo cp cudnn.h /usr/local/cuda/include
 
 
  • 1

再cd切换进lib64文件夹,执行如下命令:

sudo cp lib* /usr/local/cuda/lib64/
 
 
  • 1

3)进入复制后的动态链接库进行新的链接:

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.6
sudo ln   -s  libcudnn.so.6.0.21  libcudnn.s.6
sudo ln   -s  libcudnn.so.6  libcudnn.so
 
 
  • 1
  • 2
  • 3
  • 4

注意:此处Cudnn版本不一样,后面的序号也会有差别,建议用补全键进行文件补全。
4)设置环境变量和动态链接库:

sudo gedit /etc/profile
 
 
  • 1

在打开的文件末尾加上(“=”前后不要有空格)

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

保存之后创建链接文件:

sudo vim /etc/ld.so.conf.d/cuda.conf  
 
 
  • 1

按下键盘i进行编辑:

/usr/local/cuda/lib64
 
 
  • 1

按下Esc键,输入:wq保存退出。
最后在终端输入以下命令使该链接生效:

sudo ldconfig
 
 
  • 1

5)使用sample里面的例子来测试还需要编译。进入用例文件进行编译(注意执行第二部命令时需要较长时间是正常现象):

cd /usr/local/cuda/samples  
sudo make all -j4  
cd /usr/local/cuda/samples/bin/x86_64/linux/release  
sudo ./deviceQuery  
 
 
  • 1
  • 2
  • 3
  • 4

执行命令后出现如下信息,则说明显卡驱动和cuda安装成功。
这里写图片描述

三、TensorFlow安装
直接运行命令:

sudo pip install pip --upgrade  //对pip进行版本升级
sudo pip install tensorflow-gpu  //安装GPU版的tensorFlow
 
 
  • 1
  • 2

确认是否安装成功:

python
import tensorflow as tf
 
 
  • 1
  • 2

如下图片:
这里写图片描述

注意,Ubuntu16.04,默认pip版本为8.0,而TensorFlow安装需要版本为pip 9.0.1
另外,由于pip安装为国外文件下载,所以下载速度会很慢,在安装过程中,注意换源或则执行如下方法:
报错如下:
ReadTimeoutError: HTTPSConnectionPool(host=’pypi.python.org’, port=443): Read

解决办法一:
加大超时时间:pip --default-timeout=100 install -U pip

解决办法二:
在~目录下,建一个.pip的目录:

mkdir .pip
cd .pip
gedit pip.conf
 
 
  • 1
  • 2
  • 3

在文件中写入以下内容,保存并关闭(选用的豆瓣源):

[global]
index-url = http://pypi.douban.com/simple
trusted-host = pypi.douban.com#没有这句会包warning
disable-pip-version-check = true #版本不检查
timeout = 120#超时时间设置 
 
 
  • 1
  • 2
  • 3
  • 4
  • 5

四、caffe安装
1.安装各种依赖包

sudo apt-get update  
sudo apt-get install -y build-essential cmake git pkg-config  
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler  
sudo apt-get install -y libatlas-base-dev  
sudo apt-get install -y --no-install-recommends libboost-all-dev  
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev  
sudo apt-get install -y python-pip  
sudo apt-get install -y python-dev  
sudo apt-get install -y python-numpy python-scipy  
sudo apt-get install -y libopencv-dev  
 
 
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10

2.下载caffe

cd ~
git clone https://github.com/BVLC/caffe.git    //从github上git caffe 
 
 
  • 1
  • 2

3.安装配置caffe

cd caffe    //打开到刚刚git下来的caffe  
cp Makefile.config.example Makefile.config //make指令只能make Makefile.config  
sudo gedit Makefile.config      //打开Makefile.config文件  
 
 
  • 1
  • 2
  • 3

4.配置文件修改(看英文提示根据具体修改):
1)应用cudnn

#USE_CUDNN :=1
修改成: 
USE_CUDNN := 1
 
 
  • 1
  • 2
  • 3
  • 4

2)应用opencv版本

#OPENCV_VERSION := 3 
修改为: 
OPENCV_VERSION := 3
 
 
  • 1
  • 2
  • 3
  • 4

3)使用python接口

#WITH_PYTHON_LAYER := 1 
修改为 
WITH_PYTHON_LAYER := 1
 
 
  • 1
  • 2
  • 3
  • 4

4)修改python路径:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
 
 
  • 1
  • 2

修改为:

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial       
 
 
  • 1
  • 2

5)修改caffe目录下的Makefile文件:
将:

NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
 
 
  • 1

替换为:

NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
 
 
  • 1

将:

LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
 
 
  • 1

改为:

LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
 
 
  • 1

6)最后进入caffe目录,进行编译

cd caffe
make all -j4
make test -i4
make runtest
make pycaffe
 
 
  • 1
  • 2
  • 3
  • 4
  • 5

正确结果如下:
这里写图片描述

如果编译没有报错,则基本没有问题。

7)相关库安装:
首先要安装python接口依赖库,在caffe根目录的python文件夹下,有一个requirements.txt的清单文件,上面列出了需要的依赖库,安装这个清单安装就可以。
在安装scipy库的时候,需要fortran编译器,如果没有这个编译器就会报错,因此,可以先安装以下。
首先进入 caffe/python 目录下,执行安装代码:

sudo apt-get install gfortran
for req in $(cat requirements.txt); 
do sudo pip install $req; 
done
 
 
  • 1
  • 2
  • 3
  • 4

之后,执行以下命令:

sudo pip install -r requirements.txt
 
 
  • 1

就会看到,安装成功的,都会显示Requirement already satisfied, 没有安装成功的,会继续安装。

注意如果有如下问题:
No module named caffe,
解决办法一
终端中输入

sudo gedit /etc/profile
 
 
  • 1

在最后添加:

export PYTHONPATH=/home/yp/caffe/python:$PYTHONPATH
 
 
  • 1

保存退出后,在终端输入:

source /etc/profile
 
 
  • 1

解决办法二:

sudo gedit ~/.bashrc

#add the folling line

export PYTHONPATH=/path/to/caffe-master/python:$PATHONPATH

source ~/.bashrc
 
 
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

解决办法三:
每次写代码的时候,都直接将caffe的路径加入即可,只要caffe安装成功,其他都是小问题

import sys
sys.path.append('/path/to/caffe-master/python')
import caffe
 
 
  • 1
  • 2
  • 3

8)安装jupyter:

sudo pip install jupyter
 
 
  • 1

安装完成后,运行notebook

jupyter notebook
 
 
  • 1

ipython notebook
 
 
  • 1

出现如下画面:
这里写图片描述

9)验证是否安装正确:

python
import caffe
 
 
  • 1
  • 2

转载自https://blog.csdn.net/u013401766/article/details/78559495

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值