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

Question

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

我可以访问hpc，内存资源很高，但它没有启用GPU。

Answer 1

是的，您可以完全在CPU上运行它。为此，你只需要做一个小小的改变：

打开文件train.py并包含行

os.environ["CUDA_VISIBLE_DEVICES"]=""

在包含tensorflow之前。

Answer 2

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