1.    配置Anaconda3

1.1.  下载安装包

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu

1.2.  安装

下载完之后是后缀.sh文件

在终端输入:bash Anaconda3-5.0.0.1-Linux-x86_64.sh

2.    切换默认Python版本

2.1.  查看Python版本

终端输入python查看版本,如果默认不是python3,则在终端输入:

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150  

执行完上述两行命令之后在终端输入python,如果第一行信息是3.6,则成功。效果图如下:

python检查GPU可用时显示cpu python gpu运行_ubuntu_02

如果要切换到Python2,执行:

sudo update-alternatives --config python

2.2.  添加环境变量

将Python添加到环境变量中。

如果在安装Anaconda的过程中没有将安装路径添加到系统环境变量中,需要在安装后手工添加。

# 打开profile文件

$sudo gedit /etc/profile

# 在文件末尾处添加

exportPATH=/home/liguangyao/anaconda3/bin:$PATH

其中,将“/home/liguangyao/anaconda3/bin”替换为你实际的安装路径,然后保存。

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_03

2.3.  使环境变量生效

让etc/profile文件修改后立即生效,有以下三种方法:

# 方法一 (注意:.和/etc/profile之间有空格)

. /etc/profile 

# 方法二:在终端输入

source /etc/profile

# 方法三:在终端输入

source ~/.bashrc

然后重启系统,打开终端,输入Python,出现以下界面,表明设置成功。

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_04

也可以在终端输入ecoh $PATH查看已有的环境变量 ,确认输出是否已经有Anaconda路径了。

3.    安装Pycharm

参考Ubuntu安装Pycharm并添加系统快捷启动图标

4.    Cuda9.0安装

4.1. NVIDIA驱动安装

Ubuntu18.04自带了NVIDIA驱动,但是它没有安装完整,不能在终端使用nvidia-smi命令查看,在之后的CUDA编译测试里面也会出现问题,因此需要重新安装。重新安装的方式如下:

# 首先,检测你的NVIDIA图形卡和推荐的驱动程序的模型

ubuntu-drivers devices

python检查GPU可用时显示cpu python gpu运行_ubuntu_05

从中可以看到,这里有一个设备是Quadro K2000,对应的驱动是NVIDIA -390和340,系统推荐的安装390版本的驱动。有两种方法:

方法一、选择安装所有推荐的驱动

sudo ubuntu-drivers autoinstall

方法二、只安装其中一个驱动

sudo apt install nvidia-390

笔者选择第一种方式,执行完命令后,在终端输入:nvidia-smi 可以得到相关信息。

python检查GPU可用时显示cpu python gpu运行_ubuntu_06

4.2. 降级GCC

因为Ubuntu18.04默认gcc7.0,而CUDA9.0只支持gcc6.0及以下版本,因此需要降级,笔者选择降级到的版本是gcc5.5版本。

# 首先查看自己的版本

gcc –version

笔者机器显示7.3.0,因此降级版本并激活。

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_07

# 在终端输入

sudo apt-get install gcc-5 g++-5

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 50

# 这时候再在终端输入

gcc –version

如果显示是5.5版本则OK。如果出现其他错误未降级情况,请自行百度/Google查看解决方案。

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_08

4.3. 依赖库的安装

# 在终端输入

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev

sudo apt-get install libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

python检查GPU可用时显示cpu python gpu运行_ubuntu_09

成功安装完依赖库!

4.4. 安装Cuda9.0

官网地址:点击此处进入

python检查GPU可用时显示cpu python gpu运行_ubuntu_10

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_11

下载后存放的目录需要记住。

#在安装包存放目录处打开终端输入

Sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_12

sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_13

sudo dpkg -i cuda-repo-ubuntu1704-9-0-local-cublas-performance-update_1.0-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu1704-9-0-local-cublas-performance-update-2_1.0-1_amd64.deb

python检查GPU可用时显示cpu python gpu运行_Python_14

Sudo apt-get update

#这一步可能会出现如下错误:

The following packages have unmet dependencies:

cuda : Depends: cuda-9-0 (>= 9.0.176) but it is not going to beinstalled

E: Unable to correct problems, you have held broken packages.

解决方法参见博客:Ubuntu18.04 LTS下Caffe-GPU版本安装常见错误

#在终端继续执行:

Sudo apt-get install cuda

python检查GPU可用时显示cpu python gpu运行_python_15

在/usr/local目录下可以看到cuda和cuda-9.0两个文件夹,表面安装成功,接下来进行环境变量配置。

4.5. 环境配置

完成以上的步骤以后一定要进行环境的配置。步骤如下:

#打开终端输入

sudo gedit ~/.bashrc

# 会弹出一个可写的配置文件,在末尾把以下配置写入并保存

export CUDA_HOME=/usr/local/cuda-9.0

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATHs

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

export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/lib64"

python检查GPU可用时显示cpu python gpu运行_Python_16

# 最后执行

source ~/.bashrc

重启电脑。

4.6. 安装测试

在安装的时候也相应安装了一些cuda的一些例子,可以进入例子的文件夹然后使用make命令执行。

从网上找了两个例子,例一:

# 第一步,进入例子文件

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery

# 第二步,执行make命令

sudo make

# 第三步,运行Demo

./deviceQuery

如果结果有GPU的信息,说明安装成功。

python检查GPU可用时显示cpu python gpu运行_Python_17

例二:

# 进入例子对应的文件夹

 cd /usr/local/cuda/samples/5_Simulations/fluidsGL

# 执行make

 sudo make clean && sudo make

# 运行

 ./fluidsGL

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_18

自带例子测试通过,Cuna9.0安装配置完成!

5.    CuDnn7.0安装

官网下载地址:点击这里(需要注册个账号):

为适应Cudn9.0,笔者下载的是CuDnnv7.0.4 (Nov 13, 2017), for CUDA 9.0。笔者下载完的CuDnn是.solitairetheme8格式的文件。

python检查GPU可用时显示cpu python gpu运行_ubuntu_19

python检查GPU可用时显示cpu python gpu运行_python_20

# 解压

tar -xvzf cudnn-9.0-linux-x64-v7.solitairetheme8

python检查GPU可用时显示cpu python gpu运行_ubuntu_21

# 在终端输入

sudocp cuda/include/cudnn.h   /usr/local/cuda/include

# 注意,解压后的文件夹名称为cuda ,将对应文件复制到 /usr/local中的cuda内

sudocp cuda/lib64/libcudnn*    /usr/local/cuda/lib64 

sudochmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

python检查GPU可用时显示cpu python gpu运行_Python_22

此处Cudnn安装完成.

# 在终端输入(此处的V是大写)

nvcc –V

python检查GPU可用时显示cpu python gpu运行_python_23

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_24

安装成功!(若没有显示,可以重启电脑)

6.    Caffe-GPU安装

# 在终端输入

sudo apt install caffe-cuda

6.1.基本依赖库的安装

# 在终端输入

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev

sudo apt-get install libhdf5-serial-dev protobuf-compiler

sudo apt-get install --no-install-recommends libboost-all-dev

sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

6.2.配置

# 首先我们要从GitHub的远端下载caffe的源码

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

python检查GPU可用时显示cpu python gpu运行_python_25

python检查GPU可用时显示cpu python gpu运行_python_26

cd caffe sudo cp Makefile.config.example Makefile.config

sudo gedit Makefile.config

进入caffe根目录,用编辑器打开Makefile.config文件。笔者选择SublimeText。

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_27

6.2.1.修改Makefile.config的配置文件

# 将第5行注释去除

USE_CUDNN:= 1

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_28

# 将第21行注释去除

 OPENCV_VERSION := 3

python检查GPU可用时显示cpu python gpu运行_python_29

将第37和38行注释或者删除

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_30

修改后如图所示

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_31

将第51行BLAS:= atlas注销,换成BLAS := open

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_32

将Python2环境注销,换成Anaconda3下的Python环境

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_33

将PYTHON_LIB:= /usr/lib注释

取消PYTHON_LIB:= $(ANACONDA_HOME)/lib的注释

python检查GPU可用时显示cpu python gpu运行_python_34

INCLUDE_DIRS :=$(PYTHON_INCLUDE) /usr/local/include /usr/local/hdf5/include

LIBRARY_DIRS :=$(PYTHON_LIB) /usr/local/lib /usr/lib /usr/local/hdf5/lib

python检查GPU可用时显示cpu python gpu运行_Python_35

删除以下黄色背景操作
6.2.2.修改Makefile文件

在终端输入:sudo gedit Makefile

# 这行去掉  

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

# 添加这行

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

python检查GPU可用时显示cpu python gpu运行_python_36

6.3.    Caffe源码中安装Python的必要项

# 在终端输入

cd /home/liguangyao/programming/caffe/python 

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

6.4.   编译

# 进入caffe的根目录下

cd /home/liguangyao/Programming/caffe

sudo make clean

sudo make all -j4 (-j4表示使用4核处理器执行当前指令)

python检查GPU可用时显示cpu python gpu运行_python_37

sudo make test -j4 (最好加上sudo防止有些文件的访问权限不够)

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_38

sudo make runtest -j4  


此步可能会报错,解决方法参见博客:Ubuntu18.04 LTS下Caffe-GPU版本安装常见错误

python检查GPU可用时显示cpu python gpu运行_python_39

sudo make pycaffe -j4(配置pycaffe)

python检查GPU可用时显示cpu python gpu运行_Python_40

返回上一目录,输入Python测试

python检查GPU可用时显示cpu python gpu运行_ubuntu_41

Congratulations!Caffe-GPU编译成功!

7.    MNIST 模型

#进入Caffe根目录

 cd /home/liguangyao/Programming/caffe

7.1. 数据集准备

#下载数据集

sudo ./data/mnist/get_mnist.sh

#转换数据格式

sudo ./examples/mnist/create_mnist.sh

7.2. 数据集训练

#训练

sudo ./examples/mnist/train_lenet.sh

#训练完成后如下所示:

python检查GPU可用时显示cpu python gpu运行_python检查GPU可用时显示cpu_42

7.3. 数据集测试

有三种接口可以进行数据集测试,分别是命令行、Python和MATLAB,笔者采用命令行进行测试。

在在caffe根目录下创建文件。在终端输入:

touch test_lenet.sh

sudo gedit test_lenet.sh

在文件中添加:

./build/tools/caffe.bin test -modelexamples/mnist/lenet_train_test.prototxt -weightsexamples/mnist/lenet_iter_10000.caffemodel -iterations 100

#在终端执行

sudo sh test_lenent.sh

python检查GPU可用时显示cpu python gpu运行_ubuntu_43

Congratulations!测试成功!至此,Caffe-GPU版本安装测试完毕!

可以尽情的开启Caffe深度学习之旅了!