栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

tensorflow2版本安装的一些坑

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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降级后,再次实验,发现没有报错出现了。

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/998521.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号