需求

在家呆的时间打算学习一下AI，在网上查了挺多机器学习的，最后选择了TensorFlow。主要目的是给自己生成一套神经网络，留着以后用。

基础

本身电脑Win10 1660显卡一张。

环境

Python: 3.8.1
TensorFlow-GPU: 2.6.0
CUDA: 11.4.1
CUDNN: 11.4

环境安装过程

1. Python

python是开发人员的必备品，已经安装过了，这里就不做介绍了，自行百度。

1. TensorFlow-GPU

开源的计算机学习平台，安装比较简单，python下的pip可直接安装：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade tensorflow-gpu

命令解析：

pip python包管理器
install 安装
-i 临时指定安装源，这里选的是清华的源，要不下载太慢
--upgrade 顺带更新
tensorflow-gpu 软件包名称

然后尝试使用tensorflow，python命令行模式下引入此库。

import tensorflow as tf

没有安装CUDA和CUDNN的情况下会出现报错，如下图：

1. 安装CUDA
   进入英伟达开发者中心下载：CUDA Toolkit Archive 或 阿里云盘

下载后点击会先把安装文件解压到指定目录，之后才会真正的安装。

安装图示：

1. 安装CUDNN
   进入英伟达开发者中心下载：cuDNN Archive

下载后需要解压到CUDA的安装目录中。

1. 环境变量配置
   右键此电脑-》高级系统设置（右边小字）-》高级-》环境变量-》系统变量Path

检查变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp
C:\Program Files\NVIDIA Corporation\Nsight Compute 2021.2.1\
// C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64

头三个是安装完CUDA时候自动添加，最后一个需要手动添加

最后所有环境处理完毕，入门就是Hello World

代码：

import tensorflow.compat.v1 as tf   #解决 module ‘tensorflow’ has no attribute ‘Session’
#import tensorflow as tf

tf.compat.v1.disable_eager_execution()  #解决报错：runtimeerror: the session graph is empty. add operations to the graph before calling run()
hello = tf.constant('Hello world')

sess = tf.Session()

with tf.Session() as sess:
    print(sess.run(hello).decode())

sess.close()