一步一步教你利用Anaconda配置安装Tensorflow2.1

前言

对于tensorflow小白的我，终于安装上了tensorflow2.x
现在(2020/6/29)电脑版本号对应：
tensorflow2.0-gpu --Python3.6-- CUDA10.0— cuDNN 7.5
Windows 安装tensorflow2.0时，必须使用Python3.6
Tensorflow2.0，驱动版本要410.x以上 只能使用 CUDA10.0
以下步骤较详细，可供参考！

准备

1.查看电脑有独立显卡（是否支持GPU加速）

电脑小白如果对自己电脑部了解，可以查看是否有NVIDIA面板

如果有则说明拥有显卡，如若没有，NVIDIA面板则使用鲁大师进行检测和更新驱动

驱动不满足，可使用鲁大师进行更新 或进入官网：https://www.nvidia.com/Download/index.aspx?lang=en-us

2.检查显卡可支持的CUDA版本

Tensorflow2.1，则使用 CUDA10.1

一、下载Anaconda

1.下载

1. 官网下载
   链接：https://www.anaconda.com/
2. 云盘下载：
   链接：https://pan.baidu.com/s/1knreO78v3bh4QeOm7SahMQ
   提取码：i86y
   
   

2.安装Anaconda

这一步根据个人需求，我是个人开发使用，默认选择了第一个

选择安装路径时注意，不要带中文，空格等非法字符，会报错

注意：勾选第一个，把Anaconda加入环境变量

等待


Anaconda3 安装成功！！

二、安装Tensorflow2.1

1.打开Anaconda Prompt

2.创建Anaconda环境（Python3.7）

conda create -n tf2.1 python==3.7

3.输入"y"

输入y下载各种包

4.进入已创建的环境

conda activate tf2.1

5. 安装英伟达 SDK

具体版本对应关系：https://blog.csdn.net/qq_27825451/article/details/89082978

TF版本	Python 版本	CUDA	cuDNN
tensorflow_gpu-1.15	2.7、3.3-3.6	CUDA Toolkit 10.0	cuDNN v7.5
tensorflow_gpu-2.0.0	2.7、3.3-3.7	CUDA Toolkit 10.0	cuDNN v7.6
tensorflow_gpu-2.1.0	2.7、3.3-3.7	CUDA Toolkit 10.1	cuDNN v7.6

1）命令行安装

这里选择英伟达 SDK10.1版本

conda install cudatoolkit=10.1

选择 “y” 安装各种包

2）手动下载安装

步骤网上较多，此处不详细介绍

官网（需登录注册）：https://developer.nvidia.com/rdp/cudnn-download
云盘下载：
链接：https://pan.baidu.com/s/1Qal4Mt-8Hd-tfIvJEYiEJw
提取码：afj5

下载完安装后，查看环境变量，是否有以下三处：

6.安装英伟达深度学习软件包

这里选择英伟达软件包7.6版本

1）使用命令行

conda install cudnn=7.6

选择 “y” 安装各种包

2）手动下载安装

三、安装Tensorflow

CPU版本：pip install tensorflow==2.1
GPU版本：pip install tensorflow-gup ==2.1

网速太慢使用国内镜像

pip install -i +镜像+包名

清华：https://pypi.tuna.tsinghua.edu.cn/simple
阿里云：http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学：http://pypi.hustunique.com/
山东理工大学：http://pypi.sdutlinux.org/
豆瓣：http://pypi.douban.com/simple/

如使用清华镜像

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==2.1

正在安装，稍等

四、验证是否安装成功

1.进入Python

python

出现版本号3.7，Python安装成功！！

2.引入tensorflow包

import tensorflow as tf

1）报错

ImportError: DLL load failed: 找不到指定的模块。
很可能CPU与TF版本不兼容，需移除原环境，重新安装新版本环境
参考链接：https://blog.csdn.net/qq_37960402/article/details/89022877
移除环境
在anaconda prompt中使用命令

conda remove -n 环境名 --all

移除tensorflow

2）正常

3.查看tensorflow版本

tf.__version__

安装成功！！！

五、安装PyCharm

1.下载

• 官网：

https://www.jetbrains.com/pycharm/download/#section=windows

下载社区版PyCharm

• 云盘：

链接：https://pan.baidu.com/s/13rvvBtxOTWu6pcOJJQVjBw
提取码：zye8

2.安装

Next

勾选环境变量

Install

等待安装

选择重启电脑

3.创建工程

打开并创建工程

设置工程路径

设置环境变量

使用conda环境–>
选择tf2.1中的python作解释器

Creat

安装成功！！

六、在Pycharm中测试Tensorflow

新建test.py测试文件
输入代码

import tensorflow as tf
tensorflow_version = tf.__version__
gpu_avaliable = tf.test.is_gpu_available()

print("tensorflow version",tensorflow_version,"\tGPU avaliable:",gpu_avaliable)

a = tf.constant(1.0,2.0,name="a")
b = tf.constant(1.0,2.0,name="b")
result = tf.add(a,b,name="add")
print(result)

运行，若控制台出现

Tensor("add:0", shape=(), dtype=float64)

则Tensorflow环境配置成功！！！！