Tensorflow-GPU 2.1 &1.14版本安装教程

注意!内容记录于2020年4月8日,本文使用Anaconda3(conda 4.5.11),对应Python版本3.7进行安装。

Anaconda版本在pompt下输入可查:

conda -V

接下来的安装内容可以在Tensorflow官方网站上找到,点击左上角安装,就会是官网教程页面。
在这里插入图片描述
接下来的安装分为两个部分:

  1. 硬件配置
  2. 软件安装
硬件配置

要点

  • CUDA必须与cuDNN版本一致,两个都要装
  • 将要下载的TF版本必须支持该CUDA版本

点击左侧GPU板块,可以看到必须的软硬件要求
在这里插入图片描述显卡算力不必多说,软件要求中,GPU驱动CUDA10.1,一般CUDA10.0对应TF1,CUDA10.1对应TF2,大家按照自己的需求来装,在NVIDIA Control Panel->帮助->系统信息->组件->Nvidia DLL查看自己的驱动版本。工具包附带的已经有了。
我两个版本都装了,实际使用的是CUDA10.1,通过后续安装TF1发现10.1也能用。
在这里插入图片描述
接下来下载cuDNN,下载完压缩包以后根据NVIDIA官网安装教程,移动一些文件就OK。

软件安装TF2

要点

  • 首先建立虚拟环境,然后在虚拟环境中进行包的安装
  • 刚装完,解释器是不能使用包的,要在虚拟环境下重装各种解释器,推荐!!用Anaconda Nevigator安装

第一个板块列举了tensorflow2的安装环境,对于windows用户来说只要注意是>=py3.5,以及>=WIN7
在这里插入图片描述
然后进入安装部分,点击左侧pip板块网址

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值