注意!内容记录于2020年4月8日,本文使用Anaconda3(conda 4.5.11),对应Python版本3.7进行安装。
Anaconda版本在pompt下输入可查:
conda -V
接下来的安装内容可以在Tensorflow官方网站上找到,点击左上角安装,就会是官网教程页面。
接下来的安装分为两个部分:
- 硬件配置
- 软件安装
硬件配置
要点
- 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板块网址,