TensorFlow小记

mingxiang说，作业需要用TensorFlow 2.6.0才能保证计算图匹配。

 

装好所有包后，报错很多次，折腾老半天，最后发现是包的版本不匹配

 

TensorFlow 2.6.0匹配的numpy和keras版本

• numpy 1.19.2
• keras 2.6.0

 

tensorflow-gpu 2.6版本安装及cuda、cudnn配置_tensorflow2.6-gpu分配-CSDN博客

tensorflow_gpu-2.6.0对应版本

CUDA 11.2

cuDNN 8.1

（注意：TensorFlow 2.6.0不支持CUDA 12.6及对应cuDNN）

 

安装过程参考该链接，注意版本

cuda 和 cudnn 库的卸载与安装 - 知乎

CUDA安装以及cuDNN配置_cudnn8.1.0-CSDN博客