tensorflow - 我可以完全在 CPU 上运行 deeplab 吗？

Question

我可以完全在 CPU 上运行 DeepLab 图像分割吗？

我可以访问具有高内存资源的 hpc，但它不支持 GPU。

Answer 1

是的，您可以在 CPU 上完全运行它。为此，您只需要进行一些小的更改：

打开文件 train.py 并包含该行

os.environ["CUDA_VISIBLE_DEVICES"]=""

在包含张量流之前。

Answer 2

是的你可以。事实上，您可能根本不需要更改代码就可以在只有 CPU 的机器上运行它。