最友好的安装方式:就是利用 Anaconda
1、上tensorflow官网查看自己要安装的tensorflow版本,以及该版本对应的python版本。如下图所示。
本文安装的是tensorflow-gpu-2.1,所以要求python版本在3.5-3.7。本文python版本为3.6
2、创建环境
conda create --name tf2 python=3.6
"注":
国内下载tensorflow可能会很慢,可以添加国内的镜像源,如清华的源,加快下载速度。具体添加方法可以参考我的另一篇博客。镜像源安装第十条10,直接复制命令执行即可
3、启动刚才创建的python3.6环境,并安装tensorflow
conda activate tf2
conda install tensorflow-gpu=2.1
通过上边的命令即可自动安装对应的版本cuda和cudnn。从上图中可以看到要求cuda=10.1,cudnn=7.6,在执行了安装tensorflow命令之后,会弹出即将安装的所有包,可以看到,cuda和cudnn也是满足上图中的版本要求。如下图所示。
4、测试是否安装成功
进入刚才新建的tf2环境的python中
import tensorflow as tf
tf.config.experimental.list_physical_devices('GPU')
会弹出如下图所示信息。
说明安装成功!