最近更新:2020年4月2日
首先,安装Tensorflow时,版本一定要正确,这里有Python版本,Cuda,Cudnn的版本等。如果一个版本不对,很容易出错。以前tensorflow要求python3.5的时候,我用了3.6,结果后面就麻烦了。后来,我就学会了老老实实看官方文档,只安装对应的版本。
一切软件的安装,都应该以官方文档为准。本文仅仅看作一篇官方文档的阅读指南。我个人的习惯是,首先看官方文档,看不懂查CSDN。
Python版本和环境
首先,进入安装页面,查看最新的安装指南:
https://tensorflow.google.cn/install
看看你的python版本是否符合要求,不如不符合,更改python版本,或者新建虚拟空间,并在虚拟空间安装符合要求的python版本。
官方建议安装在虚拟环境。
GPU
打开GPU页面,
https://tensorflow.google.cn/install/gpu
找到软件要求(Software requirements)如下:
由图可见,Tensorflow要求Cuda 10.1,所以,我们坚决不装10.2。
我们严格按照要求,依次安装Nvidia驱动,Cuda和Cudnn。
Nvidia驱动,我每次都是安装最新版的。Cuda和Cudnn,只安装要求安装的版本。
Cudnn的安装和Windows设置,跳到Windows设置部分:
https://tensorflow.google.cn/install/gpu#windows_setup
安装Tensorflow-GPU
一切都准备就绪了,最后一步就比较容易了。
pip install tensorflow-gpu
升级:
pip install tensorflow-gpu --upgrade
清华的Conda镜像,居然不支持tf2.0: