Tensorflow使用cpu(tensorflow)与gpu(tensorflow-gpu)后端显示不同的行为

时间:2019-02-05 19:36:57

标签: python tensorflow gpu

我正在使用利用张量流的代码(https://github.com/vuptran/cardiac-segmentation)。

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

但是,如果我使用纯张量流,它运行时不会出错,但是的行为却有所不同(在该it produces completely nonsensical results中)。

1。 1 1 1 >

2。或者,是否有一种方法可以安装tensorflow-gpu后端,但是可以通过某种方式 hack 将其从CPU运行? 2


1。这个has been discussed in the bug tracker previously,但作者说代码不是特定于gpu的,并且不知道使用cpu后端时tensorflow行为不同的任何明显原因。
2。简单地设置CUDA_ENABLED_DEVICES=''是行不通的,因为代码在尝试导入tensorflow时将失败,因为tensorflow会引发关于无法找到相关库的错误。

0 个答案:

没有答案