caffe安装_1.ubantu+cuda+cudnn安装

最新推荐文章于 2022-03-18 21:50:06 发布

is_fight

最新推荐文章于 2022-03-18 21:50:06 发布

阅读量260

点赞数

分类专栏： caffe学习文章标签： ubuntu16.04 cuda8.0 cudnn5.1 caffe环境配置 ubuntu1604 cuda cudnn

本文链接：https://blog.csdn.net/vahalla233/article/details/81120722

版权

caffe学习专栏收录该内容

7 篇文章 0 订阅

订阅专栏

首先安装cuda依赖项

sudo apt-get update
sudo apt-get upgrade #升级软件后再安装依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install git cmake build-essential

配置一些基本的环境变量vim ~/.bashrc加入：

export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH

保存退出然后source .bashrc

更改grub：

sudo gedit /etc/default/grub

找到:GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
修改为：GRUB_CMDLINE_LINUX_DEFAULT="quiet splash nomodeset"
更新GRUB：sudo update-grub

为后续安装软件方便更换gcc版本（有些是要求complier版本的）使用update-alternatives工具设置gcc的快捷切换：

sudo apt install gcc-5 g++-5
sudo apt install gcc-6 g++-6
 
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 50 --slave /usr/bin/g++ g++ /usr/bin/g++-5
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 60 --slave /usr/bin/g++ g++ /usr/bin/g++-6
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 70 --slave /usr/bin/g++ g++ /usr/bin/g++-7
 
sudo update-alternatives --config gcc //会出现让你选择哪个版本

一、安装显卡驱动

驱动版本要高于cuda后面比如声明的cuda_9.2.88_396.26_linux.run的396.26

第一步：关闭图形界面，ctrl+alt+F2切换到文本界面操作。

sudo service lightdm stop   #ubuntu16.04关闭图形界面，开启stop改start
sudo service gdm stop       #ubuntu18.04关闭图形界面，开启stop改start

安装前关闭自带的开源驱动nouveau ：

sudo vim /etc/modprobe.d/blacklist.conf
//在打开的文件中最后一行输入
blacklist nouveau
options nouveau modeset=0

可以查看推荐driver（但实际所选版本需要适用于cuda版本）：终端窗口输入：

ubuntu-drivers devices

卸载掉原有nvidia驱动

sudo apt-get remove --purge nvidia-* #卸载通过源码安装的
sudo autoremove
（通过run文件下载的 sudo /usr/bin/nvidia-uninstall卸载cuda：sudo /usr/local/cuda/bin/uninstall_cuda_9.2.pl）

接下来正式安装！

方法1：ctrl+alt+F2进入ttf2界面，

sudo apt-get install nvidia-367   nvidia-prime #看自己需要哪个版本nvidia-？
sudo nvidia-xconfig

方法2：nvidia官网下驱动，下载完成后，到该目录下执行：

sudo chmod + x NVIDIA-Linxu-x86_64-版本号.run  #获取权限
sudo ./NVIDIA-Linxu-x86_64-版本号.run  -no-x-check -no-nouveau-check -no-opengl-files #没有no双显卡系统会循环登陆

提示步骤Accept->Continue installation->sign the kernel module（为内核模块签名）？选择是->使用已有的密钥给模块签名还是重新生成？选重新生成密钥->是否删除已生成的密钥文件？选择否->安装完成。

最后，为了能够编译CUDA提供的例子，还需要再安装如下包：sudo apt-get install freeglut3-dev

安装好后建立软链接

sudo ln -s /usr/lib/x86_64-linux-gnu/libglut.so.3 /usr/lib/libglut.so
sudo reboot或shutdown -r now

重启后nvidia-smi和nvidia-settings验证是否安装成功
安装失败可以尝试安装依赖项：

sudo apt-get install lib32ncurses5
sudo apt-get install lib32z1

二、安装CUDA

cuda官网下载所需要的runfile文件，注意版本需要和你要安装的caffe或tf等深度学习框架的版本适应！比如tf1.4对应cuda8.0

sudo service lightdm stop #ubuntu16.04关闭图形界面
sudo service gdm stop#ubuntu18.04关闭图形界面

cd到下载的cuda_******.run文件目录下

sudo sh  ./cuda_9.2.88_396.26_linux.run  --no-opengl-libs
#没有no双显卡系统会循环登陆，相反单显卡不要后面的--no

一堆说明文件ctrl+C略过，除了驱动项选择no，其它使用默认值即可
配置环境变量sudo gedit /etc/profile 打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

export  PATH=/usr/local/cuda-8.0/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH

保存，然后安装官方patch更新：到deb目录

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-update-1_1.0-1_amd64.deb
sudo apt-get update  
sudo apt-get upgrade cuda
或者.run安装 sudo sh ./CUDA_....run

重启电脑，进入终端

cd  /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

如果 result = PASS ，则安装成功，否则失败，可能原因是环境未配置好，或者是cuda与驱动版本不兼容。

或者nvcc -V也可以验证是否cuda配置完成！
安装完毕后，再声明一下环境变量，并将其写入到 ~/.bashrc 的尾部，sudo vim ~/.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}}

source ~/.bashrc更新环境，然后设置环境变量和动态链接库，

在命令行输入：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使链接立即生效。

三、安装cudnn

cudnn官网下载 for cuda对应版本的，cd到压缩包位置解压

sudo tar -zxvf ./cudnn-8.0-linux-x64-v5.1.tgz

进入cuDNN5.1 解压之后的include目录，在命令行进行复制头文件操作：

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

再进入lib64目录下的动态文件进行复制和链接：

cd ..
cd lib64
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo cp lib* /usr/local/cuda/lib64/   #复制动态链接库
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
#后面的步骤可能不需要
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5   #删除原有动态文件
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5  #生成软衔接（注意这里要和自己下载的cudnn版本对应，可以在/usr/local/cuda/lib64下查看自己libcudnn的版本）

sudo ln -s libcudnn.so.5 libcudnn.so   #生成软链接
sudo ldconfig -v  #必须在/usr/local/cuda/lib64/目录下执行

可能遇到的问题以及相应的解决办法：

1、每次开机后提醒系统出现错误，屏蔽的办法：sudo vim /etc/default/apport将1改为0保存退出
2、安装完驱动分辨率强制变换：sudo nvidia-settings打开 Nvidia x server settings这个程序，

然后在左侧目录选择 x server display configuration ，这时在右下退出和帮助按钮上面有个saveto x configuration file的按钮，

点击一下保存，下次打开就可自动修改分辨率了！

3、图形化界面下，修改系统登录界面为文本界面的配置方法sudo vim /etc/default/grub，然后

1. 将此行用“#”注释：GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"；
2. 将GRUB_CMDLINE_LINUX="" 修改为：GRUB_CMDLINE_LINUX="text"；
3. 将#GRUB_TERMINAL=console前的“#”号去除，即反注释该行。

保存修改后的配置文件，重新回到命令行界面

sudo update-grub
sudo systemctl set-default multi-user.target重启

最后，其实看caffe官网的安装指导文件是最靠谱的！

下一篇 anaconda安装 https://blog.csdn.net/vahalla233/article/details/81120934

is_fight

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
caffe安装_1.ubantu+cuda+cudnn安装

首先安装cuda依赖项sudo apt-get updatesudo apt-get upgrade #升级软件后再安装依赖sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get ins...
复制链接

扫一扫

专栏目录