ssd.pytorch icon indicating copy to clipboard operation
ssd.pytorch copied to clipboard
verified publisher icon amdegroot

AssertionError: Torch not compiled with CUDA enabled

Open MalteEbner opened this issue 5 years ago • 0 comments

When running the code without a CUDA Gpu and/or pytorch not compiled with it, the following error occurs: "AssertionError: Torch not compiled with CUDA enabled". The reason is that "ssd_pytorch/layers/modules/multibox_loss.py:35 sets the argument use_gpu per default to true. This can be easily fixed, see #537.

MalteEbner avatar Jan 04 '21 16:01 MalteEbner