ubuntu18.04系統安裝caffe過程

參考：https://blog.csdn.net/CAU_Ayao/article/details/83536320#1__18

可以按照參考鏈接的步驟安裝，以下為本人安裝時實際遇到的問題

1. 配置anaconda3環境

下載地址： https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

我下載的版本是 anaconda3-5.2.0

 選擇此版本因為該版本python是3.6

 

2. 安裝nvidia

終端輸入命令：nvidia-smi

挑選適合自己電腦的版本,成功顯示如下：

 

 

 

3. 降級gcc

在終端輸入,可以看當前gcc版本

gcc --version

 

 

 查看cuda10.2和gcc的對應關系

 

 我當前的版本是7.3.0，所以就沒有選擇降級

 

4. 安裝cuda

下載地址：https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal

 我選擇的版本是10.2，如下圖進行選擇

然后根據提示進行安裝。

 

 提示一直失敗 然后參考https://www.jianshu.com/p/b6c472231fc8

在命令行輸入

apt-get --purge -y remove 'cuda*'

apt-get --purge -y remove 'nvidia*'

apt autoremove -y

apt-get clean

reboot

重啟電腦，發現安裝成功

 

3. 安裝cudnn

1） 下載地址：https://developer.nvidia.com/rdp/cudnn-download

 

 

2）選擇 cudnn Library for Linux ，點擊下載

然后在下載位置打開終端 ，輸入命令

sudo tar -zxvf /home/XXX/cudnn-10.2-linux-x64-v5.1.tgz

然后在終端繼續輸入

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

注意，解壓后的文件夾名稱為cuda ,將對應文件復制到 /usr/local中的cuda內.

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

 

3）安裝測試

在終端輸入

nvcc –V

 

 解決方法：

在終端輸入

sudo gedit ~/.bashrc

然后在文件最后添加

export PATH="/usr/local/cuda-10.2/bin:$PATH"
export CUDA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:/usr/local/cuda-10.2/extras/CUPTI/lib64:$LD_LIBRARY_PATHs
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:/usr/local/cuda-10.2/extras/CUPTI/lib64:/usr/local/cuda-10.2/lib64:/usr/local/cuda-10.2/lib64"

終端輸入

source ~/.bashrc

再次輸入 nvcc --version，發現成功

 

4. 安裝opencv

問題1：https://blog.csdn.net/weixin_41053564/article/details/81254410

問題2：https://mlog.club/article/2282932

下載cmake-gui，取消 cuda、cuff 

 

取消   ENABLE_PRECOMPILED_HEADERS=OFF

 

終端輸入：pkg-config --modversion opencv

 

 

成功！麻蛋，整了一天才發現問題

其中缺少的文件 ippicv_linux_20151201.tgz 百度雲鏈接如下：

鏈接: https://pan.baidu.com/s/1SPf4hR40oWs9vwWzu-GpYQ  密碼: 2bti

5. 安裝caffe

參考鏈接： 

https://blog.csdn.net/CAU_Ayao/article/details/83536320#1__18

跟着文章的步驟做就可以

 

安裝過程中出現

/usr/local/cuda/include/math_functions.h:54:2: warning: #warning “math_functions.h is an internal header file and must not be used directly. This file will be removed in a future CUDA release. Please use cuda_runtime_api.h or cuda_runtime.h instead.” [-Wcpp]
#warning "math_functions.h is an internal header file and must not be used directly. This file will be removed in a future CUDA release. Please use cuda_runtime_api.h or cuda_runtime.h instead."
原因：cuda版本函數文件更新
解決方案：打開出問題的文件math_functions.cu文件，進行修改

//#include <math_functions.h>  // CUDA's, not caffe's, for fabs, signbit(刪除該行)
#include <cuda_runtime.h>     // CUDA's, not caffe's, for fabs, signbit(增加該行)
#include <thrust/device_vector.h>
#include <thrust/functional.h>  // thrust::plus
#include <thrust/reduce.h>

 

6. 測試caffe

 

 

 沒報錯就是成功了

 

 5. 在pycharm引用caffe

直接引用的話，會發現：會提示沒有caffe這個模塊

 

 

 

 

在首行添加你自己文件夾下caffe的目錄

import sys
sys.path.append('/home/yangyuqing/caffe/python')
import caffe

再運行會發現

 

 

點擊下載scikit-image包