我能完全在CPU上运行deeplab吗？

Question

我能在CPU上运行DeepLab图像分割吗？

我可以访问hpc，拥有很高的内存资源，但它不是支持GPU的.

Answer 1

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

打开文件train.py并包含行

os.environ["CUDA_VISIBLE_DEVICES"]=""

在包括丹索尔·弗洛之前。

Answer 2

可以，停那儿吧。实际上，您可能根本不需要更改代码就可以在只使用CPU的机器上运行它。