Ubuntu 18.04 安装 CAFFE

Ubuntu 18.04可以通过系统内置的预编译版本caffe，直接安装

1. 安装 驱动

查询官网驱动版本号：https://blog.csdn.net/justforacm/article/details/101438710

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

sudo apt-get update

sudo apt-get install nvidia-375（375是你查到的版本号）

sudo apt-get install mesa-common-dev

sudo apt-get install freeglut3-dev

2. 安装 CUDA

sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb

sudo apt-get update

sudo apt-get install cuda​

配置cuda8.0之后主要加上的一个环境变量声明，在文件~/.bashrc之后加上

 

gedit ~/.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}}

然后设置环境变量和动态链接库，在命令行输入

sudo gedit /etc/profile

在打开的文件里面加上（注意等号两边不能有空格）

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

保存之后，创建链接文件

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

在打开的文件中添加如下语句：

/usr/local/cuda/lib64

保存退出执行命令行：

sudo ldconfig

使链接立即生效。

3. 安装CUDNN

下载完cudnn后，命令行输入文件所在的文件夹 (ubuntu为本机用户名)

cd home/ubuntu/Downloads/

tar zxvf cudnn-8.0-linux-x64-v5.1.tgz #解压文件

cd进入cudnn5.1解压之后的include目录，在命令行进行如下操作：

sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件

再cd进入lib64目录下的动态文件进行复制和链接：（5.1.5为对应版本具体可修改）

sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软衔接

sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

4. 安装CAFFE http://caffe.berkeleyvision.org/install_apt.html

sudo apt install caffe-cuda
# 可使用Python接口调用
# 安装位置
locate caffe
 /usr/bin/caffe
 /usr/lib/python3/dist-packages/caffe

5. python调用caffe

#永久方法：将在/usr/lib/python3/dist-packages/caffe中的caffe文件夹复制进Anaconda的对应环境的lib/python3.6/site-packages目录下即可
#下面的方法失效
python
import sys
sys.path.append('/usr/lib/python3/dist-packages')
import caffe