我今天刚刚安装了新的 tf 和 cuda,但是当我运行以前与 tf-1.4 一起使用的代码时,在新的 tensorflow-2.1.0 和 cuda-10.1 下将无法运行
如何解决这个问题?
小智 3
为了社区的利益,在答案部分提及解决方案(即使它出现在评论部分)。
替换命令,
import tensorflow as tf
Run Code Online (Sandbox Code Playgroud)
和
import tensorflow.compat.v1 as tf
Run Code Online (Sandbox Code Playgroud)
已解决该错误。
有关从 Tensorflow 1.x 迁移到 2.x 的更多信息,请参阅此Tensorflow 教程,此升级脚本也会有所帮助。
| 归档时间: |
|
| 查看次数: |
5483 次 |
| 最近记录: |