ubuntu系统安装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

 

2. 安装nvidia

终端输入命令:nvidia-smi

挑选适合自己电脑的版本

 

3. 降级gcc

 

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

 ubuntu系统安装caffe过程

 

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

ubuntu系统安装caffe过程

 

 解决方法:

在终端输入

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,发现成功

ubuntu系统安装caffe过程

 

4. 安装opencv

问题1:https://blog.csdn.net/weixin_41053564/article/details/81254410

问题2:https://mlog.club/article/2282932

下载cmake-gui,取消 cuda cuff 

ubuntu系统安装caffe过程

 

取消   ENABLE_PRECOMPILED_HEADERS=OFF

 

终端输入:pkg-config --modversion opencv

 ubuntu系统安装caffe过程

 

 成功!麻蛋,整了一天才发现问题

 

5. 安装caffe

参考链接: 

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

 跟着文章的步骤做就可以

 

6. 测试caffe

 

 

 

 

 

 


 

上一篇:基于CPU版本的Caffe推理框架


下一篇:cmake windows caffe cuda版本的切换