TensorFlow 2.3 GPU-InvalidArgumentError:断言失败:[0] [Op:Assert]名称:EagerVariableNameReuse 最简单的解决方法如果您想继续使用TF 2.3.x,该解决方案

问题描述

环境:

Windows 8.1 64位 Python 3.7.7 64位 TensorFlow 2.3 Keras 2.4

CUDA:10.1 CuDNN:7

错误InvalidArgumentError: assertion Failed: [0] [Op:Assert] name: EagerVariableNameReuse

问题:在执行GPU时,出现一个简单语句model = Sequential()时出现上述错误

但是如果我在没有GPU的情况下运行脚本(未设置CUDA / CuDnn环境变量,它将按预期工作。

如此有效,我有一个GPU但不能与TensorFlow(2.3)一起使用

有人可以帮忙吗?

解决方法

从我所看到的情况来看,以下致命组合会出现此问题。

  • Windows操作系统
  • TensorFlow 2.3.x

我一直在努力解决这个问题,直到最近才发现。

最简单的解决方法

最简单的解决方案是使用TensorFlow 2.4.0rcx。它仍处于预发布状态。但是还没有给我上述问题。所以,

pip uninstall tensorflow
pip install tensorflow==2.4.0rc0

如果您想继续使用TF 2.3.x,该解决方案

如果您真的想使用TensorFlow 2.3.x,唯一的解决方案似乎就是

  • 尝试基于Linux的操作系统(Ubuntu)
  • 从源代码编译TensorFlow(根据this helpful评论)