Can this be only run on CUDA enabled pytorch?

[nla-asia]

I am getting assertion error "Torch version is not compiled for CUDA support" when try to run on macosx Mojave.

[aladagemre]

@naysumyat

A workaround:

• Open up test.py
• Comment out net.cuda() and img = img.cuda() lines
• Add map_location argument to torch.load: net.load_state_dict(torch.load(cp, map_location=torch.device('cpu')))

[opusviz]

@naysumyat

A workaround:

• Open up test.py
• Comment out net.cuda() and img = img.cuda() lines
• Add map_location argument to torch.load: net.load_state_dict(torch.load(cp, map_location=torch.device('cpu')))

Thank you. it worked.