Можно ли полностью запустить Deeplab на процессоре?

Question

Можно ли полностью выполнить сегментацию изображения DeepLab на процессоре?

У меня есть доступ к hpc, с большими ресурсами памяти, но он не поддерживает GPU.

Answer 1

да, вы можете запустить его полностью на процессоре. Для этого вам нужно только сделать небольшое изменение:

Откройте файл train.py и включите строку

os.environ["CUDA_VISIBLE_DEVICES"]=""

до включения тензорного потока.

Answer 2

Да, вы можете. На самом деле, вам может не потребоваться изменять код вообще, чтобы запустить его на компьютере с центральным процессором.