Tensorflow 1.2+Ubuntu 16.04+Cuda 8.0+cuDNN 5.1配置流程

本文详述了在Ubuntu 16.04上安装Tensorflow 1.2的完整流程,包括Cuda 8.0和cuDNN 5.1的配置。通过简单的命令行操作,指导读者正确安装和设置依赖库,确保Tensorflow GPU版本的正确运行。此外,还提供了处理显卡驱动问题的方法。
摘要由CSDN通过智能技术生成

今天刚配置了tensorflow,趁热打铁,把过程记录下来,给自己和别人一个参考。
首先安装Ubuntu16.04 。用官网的镜像用UltraISO制作启动优盘安装,具体过程不介绍了。
要安装Tensorflow,首先要安装Cuda 和cuDNN 。(当然,如果你只想用CPU训练,那就不用安装这两个包,直接跳到安装Tensorflow的位置就好啦)
官网上提供了很多种安装Tensorflow的方式,这里我们用最傻瓜的一种:
用PIP直接安装。这种安装方式优点:简单,不容易出错
缺点:太简单了,以至于依赖库的版本不能自己定制,所以,如果你要按照这个步骤安装,切记看好标题中我给出的Cuda和cuDNN的版本,别下载错了(不是这个版本识别不到) 。
如果你不是安装Tensorflow 1.2,那么一定要到Tensorflow的官网找到你安装的Tensorflow版本需要的Cuda和cuDNN版本,切勿下载了错的版本。下面开始介绍安装步骤:
首先安装Cuda:
https://developer.nvidia.com/cuda-downloads 下载需要的Cuda安装包,并且按照官网的要求输入命令即可,这里我下载的设置如下图:(大家根据自己的需要更改)
选项
按照官网,我们下载完安装包后输入以下命令完成安装:
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
下一步安装cuDNN:
https://developer.nvidia.com/cudnn 去下载cuDNN(注意版本,注意版本,注意版本)。这个要下载前要先用邮箱免费注册个账号,加入什么开发者计划。(比较简单,几分钟搞定)
下载完了之后是个压缩包。这个安装更简单,解压后会得到include 和lib64两个文件夹。安装完Cuda后,你会在/usr/local/cuda下也找到include 和lib64两个文件夹。把你解压出的文件分别复制到/usr/local/cuda下的同名文件夹中就完成了。(简单说就是Cuda少几个文件,咱们给它补充上)
这两步做完之后,将下面的东西加到环境变量:(你可以选择你喜欢的添加方式&#

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值