ubuntu18.04安装caffe

ubuntu18.04安装caffe

csdn很多安装caffe的方法都是从源码编译安装，非常麻烦，然而caffe官网上可以发现，有官方的apt安装包。

安装CUDA

在ubuntu系统上最好使用deb包来安装，如下图官网中给出了安装说明：

我们只需要执行下面给出的命令就可以了：

sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pinsudo 
mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

安装cuDNN的安装

下载需要先注册账号，注册完成后进入下载页面选择下载版本。

下载完成后使用命令安装：

sudo dpkg -i **.deb

安装完成可能需要重启

使用命令nvidia-smi来查看驱动信息。如果查看不到，尝试重启电脑。

安装opencv

也不需要下载源码编译，我们只需使用python3 的pip3来安装。

# 安装pip3
sudo apt install python3-pip

# 安装opencv的python接口,并指定版本
sudo pip3 install opencv-python==3.4.0.12
# 可能提示缺少依赖文件，根据提示安装依赖

安装caffe

在caffe官网可知，ubuntu18.04已经有官方apt安装包，只需要执行安装命令即可。

# for CPU-only version
sudo apt install caffe-cpu

#for CUDA version.  Note, the cuda version may break if your NVIDIA driver and CUDA toolkit are not installed by APT.
sudo apt install caffe-cuda

安装numpy包

# 默认numpy包的版本太高，我们需要指定安装版本
pip3 install numpy==1.15.0

测试；

$ python3                                                                                                                         
Python 3.6.8 (default, Aug 20 2019, 17:12:48)                                                                                                 
[GCC 8.3.0] on linux                                                                                                                           
Type "help", "copyright", "credits" or "license" for more information.     
>>> import caffe  
>>>