用tensorflow2运行tensorflow1.x代码的方法

问题背景:

由于tensorflow1.x最高只支持CUDA10,而30系显卡最低要求CUDA11才能运行,这导致我3090显卡没法直接运行tensorflow1.x的众多开源代码。

经过搜索和实践,发现一个好用的方法实现在30系显卡下,用tensorflow2框架运行tensorflow1.x的代码

解决方法

首先确认你安装的是tensorflow2的库,比如我用的是tensorflow-gpu2.10.1,CUDA11.7

把原先tf1.x代码里import tensorflow as tf改为以下两行代码即可

import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

这两行代码实现用tf2里对tf1旧api的兼容,并且关闭tf2的新特性,从而实现用tf2来运行tf1.x的代码

猜你喜欢

转载自blog.csdn.net/qq_28941587/article/details/128466526
今日推荐