【发布时间】：2020-02-24 22:28:02
【问题描述】：
我可以完全在 CPU 上运行 DeepLab 图像分割吗？
我可以访问具有大量内存资源的 hpc，但它不支持 GPU。
【问题讨论】：
标签： tensorflow gpu cpu semantic-segmentation deeplab
【发布时间】：2020-02-24 22:28:02
【问题描述】：
是的，您可以完全在 CPU 上运行它。为此，您只需做一点小改动：
打开文件 train.py 并包含该行
os.environ["CUDA_VISIBLE_DEVICES"]=""
在包含 tensorflow 之前。
【讨论】：
-
谢谢！这行得通。不过，我可以访问启用了 GPU 的集群。它现在工作得更快。我不能设置超过 12 的批量大小，但它仍然很快
是的，你可以。事实上，您可能根本不需要更改代码就可以在只有 CPU 的机器上运行它。
【讨论】：