Windows下安装tensorflow

安装tensorflow，在命令提示符下输入命令

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

验证tensorflow安装是否成功

import tensorflow as tf

tf.print('Hello, tensorflow!')
tf.print(tf.__version__)

各版本的区别如何？

  特别需要注意 tensorflow、CUDA、cudnn版本搭配，我安装的tensorflow是2.4.0，因此CUDA安装11.0和cuDNN安装8.0

安装CUDA

查看NVIDIA支持的CUDA版本，不能安装高于NVIDIA驱动支持的版本

 下载DUDA，下载地址:https://developer.nvidia.com/cuda-toolkit-archive，下载11.0的windows版本

安装CUDA软件，选择自定义安装，只选择CUDA，不要安装 VS 选项

安装程序通常会自动设置环境变量

验证CUDA安装是否成功，在命令提示符下输入

nvcc -V

下载cudnn，下载地址:https://developer.nvidia.com/rdp/cudnn-download，下载是一个zip压缩包，解压，将文件复制到CUDA的安装目录下，默认是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0，设置环境变量

 验证GPU是否正常使用

import tensorflow as tf

tf.print('Hello, tensorflow!')
tf.print(tf.__version__)
print('Cuda:', tf.test.is_built_with_cuda())
print('GPU:', tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))

 如果版本有兼容问题，可以关闭GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

安装完毕。