Ubuntu16.04+cuda8.0+caffe安装教程

最新推荐文章于 2024-10-05 10:09:49 发布

置顶

autocyz

最新推荐文章于 2024-10-05 10:09:49 发布

阅读量8.4w

点赞数 25

分类专栏： Ubuntu 深度学习文章标签： ubuntu cuda nvidia caffe 深度学习

本文链接：https://blog.csdn.net/autocyz/article/details/52299889

版权

本文详细介绍了在Ubuntu16.04上安装CUDA8.0、nvidia驱动、cudnn，以及配置matlab、opencv3.1.0和caffe的步骤。首先，从NVIDIA官网下载并安装适配GPU的驱动，然后安装CUDA，避免重复安装驱动。接着，测试CUDA Samples验证安装成功。再下载cudnn并进行解压、复制到CUDA相应目录。安装opencv3.1.0需要安装依赖项并编译。最后，安装caffe并配置所需依赖，完成深度学习环境的搭建。

摘要由CSDN通过智能技术生成

1、安装nvidia驱动

首先去官网上查看适合你GPU的驱动（http://www.nvidia.com/Download/index.aspx?lang=en-us）
例如，本人的GPU适合的驱动如图：
这里写图片描述

执行如下语句，安装

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-367
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev

执行完上述后，重启（reboot）。
重启后输入：

nvidia-smi

如果出现了你的GPU列表，则说明驱动安装成功了。另外也可以通过

nvidia-settings

查看自己机器上详细的GPU信息，本人机器的信息如下：
这里写图片描述

2、安装CUDA

cuda是nvidia的编程语言平台，想使用GPU就必须要使用cuda。
从这里下载cuda的安装文件
https://developer.nvidia.com/cuda-release-candidate-download
这里写图片描述
注意这里下载的是cuda8.0的runfile（local）文件。
这里是nvidia给出的官方安装指南（遇到问题时可以查阅）：
http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4HIBXnwyt
下载完cuda8.0后，执行如下语句，运行runfile文件：

sudo sh cuda_8.0.27_linux.run

执行后会有一系列提示让你确认，但是注意，有个让你选择是否安装nvidia361驱动时，一定要选择否，因为前面我们已经安装了更加新的nvidia367，所以这里不要选择安装。其余的都直接默认或者选择是即可。
安装成功后会出现如下界面：

===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Installed in /home/textminer
Please make sure that
– PATH includes /usr/local/cuda-8.0/bin
– LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64, or, add /usr/local/cuda-8.0/lib64 to /etc/ld.so.conf and run ldconfig as root
To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-8.0/bin
Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-8.0/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work.
To install the driver using this install