TensorFlow2.x得环境下如何运行TF1.x环境的代码

最近因为学习需要,所以在看别人的代码,但是发现有些教科书或是很久以前的代码都是在TF1.x的环境下编写的。而很多显卡例如Genforce系列在TF1.x中是无法派上用场的,因此需要安装TF2.x-gpu版本来运行代码,但是中间也遇到了很多问题:

系统:win11

cuda 11.2

tensorflow-gpu 2.6

下面来讲讲我的解决过程,首先在安装完对应版本的cuda和cudnn后,安装TF2.x-gpu后即可,但是会发现有诸多不兼容的地方,所以参考别人在Ubuntu环境下的解决方案给出自己在Win11环境下的流程。

首先,使用

import tensorflow as tf
import tensorflow.compat.v1 as tf
//使用上述语句来改用TF1.x

之后需要使用下述语句来禁用TF2.x新的特性

tf.disable_v2_behavior()

之后在运行时报错,“No module named tf_slim”,而tf_slim是tensorflow1.x版本中的瘦身,因此在tf1.12以上的版本中似乎没有了,因此直接使用

pip install tf_slim

成功解决后!

成功解决后,报错处会显示No module named compat,但是实际上不影响运行,以上为在TF2.x环境下运行TF1.x代码的流程。

参考文章:

(17条消息) TensorFlow2运行TF1程序并解决GPU无法使用的问题_JeremyZhao1998的博客-CSD博客​​​​​​​s

版权声明:本文为博主Satya Vaswani原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/qq_49932064/article/details/123044496

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
社会演员多的头像社会演员多普通用户
上一篇 2022年5月21日
下一篇 2022年2月24日

相关推荐