Search code examples
tensorflowgpucpusemantic-segmentationdeeplab

Can I run deeplab completely on CPU?

Can I run the DeepLab image segmentation completely on CPU?

I have access to hpc, with high Memory resources but it is not GPU-enabled.

Solution

  • yes you can run it completely on the CPU. For that you only have to make a small change:

    Open the file train.py and include the line

    os.environ["CUDA_VISIBLE_DEVICES"]=""

    before tensorflow is included.