tensorflow2版本安装的一些坑

最近使用tensorflow安装过程中遇到一些问题，这里记录一下。

我这里采用的是在conda虚拟环境中安装的tensorflow。tensorflow的版本是2.6.0，python环境使用的是python3.7

首先安装Anaconda，安装配置比较简单，这里网上有很多不再赘述。

使用conda创建一个tensorflow的虚拟环境。

conda create --name tensorflow-2.6 python=3.7

然后激活进入虚拟环境

source activate tensorflow-2.6

进入环境后进行安装tensorflow，这里我使用的是带GPU的tensorflow

pip install tensorflow-gpu==2.6.0

效果如下：

 最后安装成功后，进行测试

python3

from tensorflow import keras

from tensorflow.keras import datasets,layers,models

结果出现cannot import name 'dtensor' from 'tensorflow.compat.v2.experimental'的问题。

这里的原因是默认安装的keras版本较高，我这里默认安装的是2.9.0，这里需要给keras版本降级 ，降级为2.6.0

pip list | grep keras

pip install -U keras==2.6.0

这里降级的时候默认会删除掉高版本的keras，keras降级后，再次实验，发现没有报错出现了。