Ubuntu 16.04 安装CUDA8.0+Cudnn6.0+TensorFlow+Caffe安装

最新推荐文章于 2022-11-20 02:36:29 发布

原创

最新推荐文章于 2022-11-20 02:36:29 发布 · 置顶 · 2.4k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu #cuda #python #TensorFlow #caffe安装

本文档详细介绍了在Ubuntu 16.04上安装CUDA 8.0、CUDNN 6.0、TensorFlow和Caffe的步骤。首先，加入官方PPA源并安装NVIDIA驱动，接着安装CUDA 8.0，确保安装过程中不选择NVIDIA驱动。然后，下载并解压CUDNN，设置环境变量。之后，使用pip安装TensorFlow，并处理可能的下载超时问题。最后，安装Caffe，包括安装依赖、下载源码、配置Makefile并编译。验证所有组件安装成功后，可以开始使用这些深度学习工具。

参考博客：
Ubuntu16.04 Caffe 安装步骤记录（超详尽）：http://blog.youkuaiyun.com/yhaolpz/article/details/71375762
ubuntu16.04安装caffe：http://blog.youkuaiyun.com/u011272513/article/details/52103453#

环境说明：

1）Ubuntu 16.04：ubuntu-16.04.3-desktop-amd64.iso
2）采用CUDA 8.0 ：cuda_8.0.44_linux.run
3）Cudnn 6.0：cudnn-8.0-linux-x64-v6.0.tgz
4）TensorFlow
5）Caffe
6）Python 2.7
7）pip 9.0.1

具体安装步骤：

一：CUDA 8.0安装

（包括NVIDIA显卡驱动和CUDA库文件）
通过英伟达安装包来分别安装库的驱动程序而不是使用apt-get 来安装。如果只有CPU，则可以跳过这步。
1）打开终端，加入官方PPa源：

sudo  add-apt-repository ppa:graphics-drivers/ppa

2）按enter键确认，之后更新软件库并安装最新驱动

    sudo apt-get update
    sudo apt-get install nvidia-367 nvidia-settings nvidia-prime

Note：该Ubuntu 镜像文件在sudo apt-get update 时报错，解决方法：
输入命令：sudo apt-get purge libappstream3
之后再次执行 sudo apt-get update命令就正常了。

3)安装完成后，通过命令查看是否安装成功

     nvidia-settings

此时，安装完成后，要重启电脑，有如下效果则安装完成，否则就说明安装有问题，尝试关闭UEFI保护试试。这里写图片描述

4）从官网上下载CUDA 8.0文件，下载cuda_8.0.44_linux.run文件，将其放至~/CUDA文件夹下，cd命令进入该文件所在目录，进行MD5检验，确保安装包正常。

  sudo md5sum cuda_8.0.44_linux.run

5)检验正确后，输入如下代码安装（第一行修改文件权限，第二行执行文件安装）：

chmod 777 cuda_8.0.44_linux.run
sudo ./cuda_8.0.44_linux.run  --override

这里写图片描述

执行代码后，跳出相关信息，左下方出现–更多–(0%) 这样，
这里写图片描述
一直按Enter键直到跳到100%结束退出，接下来要求输入相关配置命令，如下图，注意别弄错了，第二行配置询问是否要安装NVIDIA图像驱动器时，一定要选n,否则系统会报错，具体如下图：

安装完成后出现如下说明安装正确：

最低0.47元/天解锁文章

新学期VIP享超值加赠