EmbedSeg
EmbedSeg copied to clipboard
Possibility to train on non-GPU machines?
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