Some errors when using non-CUDA machines

[andieich]

Hi,
I successfully installed TagLab on a Windows computer, but had some issues. I cannot use the GPU since it is made by Intel.
I therefore tried to install the CPU version of torch and tochvision. When I use the install.py script, I get this error:

append() takes exactly one argument (2 given)

This is caused by line 234. I commented out lines 232 - 236 and manually installed both packages with the following code:

pip install torch --extra-index-url https://download.pytorch.org/whl/cpu

pip install torchvision --extra-index-url https://download.pytorch.org/whl/cpu

Afterwards, I could install TagLab flawlessly.

However, when trying to run a auto segmentation, I got this error:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

So I did as proposed and changed in source/MapClassifier.py in line 98:

classifier.load_state_dict(torch.load(network_name)

to

classifier.load_state_dict(torch.load(network_name, map_location=torch.device('cpu')))

Now, everything works fine. Maybe that's something to consider for the next TagLab version.
Thanks for this amazing software!