Tensorflow使用cpu(Tensorflow)和gpu(tensorflowgpu)后端显示了不同的行为

2024-09-28 23:43:55 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用的是使用tensorflow的代码(https://github.com/vuptran/cardiac-segmentation)。在

如果我使用tensorflow gpu后端,代码可以正常工作。在

但是,如果我使用普通tensorflow,它运行时不会出错,但是的行为不同于(在这个it produces completely nonsensical results)。在

1。有什么可能的原因,为什么gpu后端会导致相同的代码从cpu后端产生不同的结果,以及在哪里寻找代码中发生这种情况的证据?1

2。或者,有没有一种方法可以安装tensorflow gpu后端,但不知何故将其从CPU运行?2


1。这是has been discussed in the bug tracker previously,但作者说代码不是特定于gpu的,也不知道为什么tensorflow在使用cpu后端时会有不同的行为。
2。简单地设置CUDA_ENABLED_DEVICES=''将不起作用,因为代码将在尝试导入tensorflow时失败,因为tensorflow将抛出一个无法找到相关库的错误。

Tags: 代码httpsgithubcomgputensorflowitcpu