ubuntu系统安装caffe过程

2021-09-24 11:49:14

参考：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

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"

终端输入