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

Question

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

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

Answer 1

是的，可以。实际上，您可能根本不需要更改代码即可在仅CPU的计算机上运行它。

Answer 2

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

打开文件train.py并添加该行

os.environ["CUDA_VISIBLE_DEVICES"]=""

在包括张量流之前。