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

[ajinkya-kulkarni]

Hi team, any update on this?

[jdeschamps]

Hi @ajinkya-kulkarni,

Sorry this repository is not really maintained, but we have plans to revamp it early next year.

In the mean time, I have merged your contributions. Thanks a lot! I guess there the get_gpu_memory() problem left, we will try to have a look.