Possibility to train on non-GPU machines?

[ajinkya-kulkarni]

Hello team, I am trying to run the 3 DSB notebooks from the examples on my macbook (no GPU), and I run into the error FileNotFoundError: [Errno 2] No such file or directory: 'nvidia-smi': 'nvidia-smi'. This part of the code exists in the EmbedSeg/utils/preprocess_data.py location, specifically def get_gpu_memory(). Is there a way to "switch-off" this get gpu memory thing for training on non-gpu machines?

Thanks, Ajinkya