Ubuntu Tensorflow-gpu安装需知与版本对照

为深度学习所用,博主预想在Ubuntu16.04上安装 显卡驱动 + CUDA + cuDNN + Tensorflow-gpu + Keras + PyCharm,参考了众多资料,最终成功将所有软件安装完毕,且能成功运行使用。该篇博客介绍了安装以上软件的前期准备工作需知,亦可通过下方链接查看其他步骤的安装教程。

前期说明及版本对照

换源及安装显卡驱动

安装 CUDA(附测试)

安装 cuDNN(附测试)

安装 Tensorflow-gpu 与 Keras(附测试)

设置 PyCharm

安装教程完整版

前期说明

1、虚拟机使用 CUDA 问题

CUDA需要使用电脑的物理显卡,而在某些虚拟机中无法调用本机的物理显卡,仅能使用其提供的虚拟显卡,因此无法使用CUDA,例如VMware,但也有某些虚拟机能够使用电脑的物理显卡,不过,虚拟机的性能相对较低,对后期实验的效率有较大影响,推荐使用本身系统为Ubuntu的电脑或安装双系统使用。博主的另一篇博文 Windows系统下,Ubuntu安装至移动硬盘(简单分析与详细安装教程)提供了在移动硬盘中安装Ubuntu系统的方法,若有需要,可以借鉴。

2、版本问题

虽然许多博客都有强调 CUDA 与 cuDNN 或与 Tensorflow 需要对应版本的问题,但经博主安装后发现,不仅仅是这三个软件需要安装对应版本,显卡驱动、CUDA、cuDNN、Tensorflow-gpu、Keras、Python、GCC 这7类都必须实现版本一一对应,否则会产生一系列似是而非的错误。具体的版本对应,博主会在本教程中列出。

3、检查电脑配置

某些博客在安装之前都强调需要检查自己电脑的一些配置,例如显卡为单显卡还是双显卡、显卡种类是否符合安装要求等。其实不同的安装方法对这些配置的要求并不相同,但以防万一,检查配置总是万无一失的。

检查显卡配置可参考:Ubuntu如何正确地切换Intel和Nvidia显卡
若上方的网址未有响应,可使用该网址:Ubuntu如何正确地切换Intel和Nvidia显卡

在终端中可使用以下命令查看电脑中有哪些显卡:

lspci -k | grep -A 2 -i "VGA"

博主个人认为该条命令无法查看电脑中的全部显卡,例如博主在终端执行该命令后的结果如下图,图中显示了Intel显卡,但博主电脑中的Nvidia显卡没有显示,但不排除博主将Nvidia显卡关闭了所以无法显示的情况。

在这里插入图片描述
若与博主出现以上相同的情况也无需担心,可使用以下命令查看电脑中是否有Nvidia显卡:

lspci | grep -i nvidia

该命令将会显示出电脑可使用的Nvidia显卡的简单信息,例如博主的:

在这里插入图片描述
而后可通过CUDA官网提供的方法,检查你的显卡能否支持CUDA的使用,官网网址:Frequently Asked Questions(将滚动条拉到最底,在该页面下方有相关回答)

在安装过程中还有其他需要查看的配置,会在而后的教程中说明,Nvidia显卡的支持保证了能否使用CUDA,如果连该条件都无法满足,也无需查看其他配置了。

4、Anaconda图形界面一站式安装

博主查阅许多资料,发现有提供 Anaconda 图形界面自动安装CUDA+cuDNN+Tensorflow的方法,博主也尝试过该方法,但不知是博主的Anaconda图形界面安装有问题,还是其他的原因,在正式安装前,Anaconda会显示将自动安装的CUDA与cuDNN版本信息,博主无论安装任何版本的Tensorflow,CUDA与cuDNN的版本都是不正确的,所以博主最终放弃了这种方法,如果有想尝试的同僚可以参考:Ubuntu搭建TensorFlow GPU环境——无错详细完整版

版本对照

在安装之前,请确定显卡驱动、CUDA、cuDNN、Tensorflow-gpu、Keras、Python、GCC的版本,最好记录在TXT文件中,防止版本安装错误,卸载重装的问题!!!

对于CUDA、cuDNN、Tensorflow-gpu、Python、GCC的版本,Tensorflow的官网提供了部分对照版本:经过测试的构建配置(将滚动条拉到最底,在该页面下方有相关表格,博主直接将其截取了一份)
在这里插入图片描述CUDA与cuDNN的版本某些仅显示了一个整数,例如cuDNN版本为7,但实际上7.4版本的cuDNN也是支持的,在cuDNN的官网中提供了某些版本的对应下载选项(官网网址:cuDNN Archive
在这里插入图片描述
在某些博客中也提供了官网未提供的其他对应版本,这里就不一一写出,需要的同僚可以自行搜索。

显卡驱动的版本是与CUDA版本对应的,在Nvidia的官网中提供了对应版本(官网网址:CUDA Toolkit and Compatible Driver Versions,拉动滚动条,可找到对应的表格,博主直接将其截取了一份)
在这里插入图片描述显卡驱动的版本号对于不太熟悉的同僚可能会觉得有点奇怪,对于该教程而言,只需记下版本的编号,代表的含义可以不用理会

Keras的版本与Tensorflow和Python相关,博主找到了一个网址,提供了其对应的关系(网址:Environments,因提供的表格内容过多,博主仅截取了一部分内容)
在这里插入图片描述
在安装之前一定要确定配置的各个版本,Python的版本博主推荐3.6,该版本也是博主安装教程中使用的版本

  • 0
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值