TensorFlow2.1——环境的安装与常见问题

神经网络的安装

首先，在Anaconda 官网下载 python3.7 版本
安装完成后，打开Anaconda命令行

运行以下命令：
1.创建TF2.1的环境

conda create -n TF2.1 python=3.7

2.进入环境

conda activate TF2.1

3 4 步安装前请确认是否有英伟达的显卡与版本是否合适
3.安装英伟达的SDK 10.1版本

conda install cudatoolkit=10.1

4.安装英伟达深度学习软件包7.6版本

conda install cudnn=7.6

5.安装TensorFlow2.1

pip install tensorflow==2.1

6.安装完成后，输入python验证是否成功

import tensorflow as tf
tf.__version__

如果显示2.10，则安装成功

7.安装pycharm与配置
官网下载pycharm安装，设置工程路径，环境变量选择 Existing interpreter 中的TF2.1

8.测试
配置好之后输入

import tensorflow as tf

tensorflow_version = tf.__version__
gpu_available = tf.test.is_gpu_available()

print("tensorflow version:", tensorflow_version, "\tGPU available:", gpu_available)

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)

成功运行并显示则配置成功

PS：如果运行失败，请确认是否为英伟达显卡最新驱动，更新驱动即可