BLINK icon indicating copy to clipboard operation
BLINK copied to clipboard

raise AssertionError("Torch not compiled with CUDA enabled") on Mac

Open fengco5520 opened this issue 4 years ago • 1 comments

I am trying to run the scripts under https://github.com/facebookresearch/BLINK#4-use-blink-in-your-codebase on Mac OS.

I am getting the errors: raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled

How to fix it? thanks

fengco5520 avatar Jan 20 '21 22:01 fengco5520

@fengco5520 It seems that you do not have access to GPU? If that's the case, for a quick fix you can change the device argument (set device='CPU') the following function: https://github.com/facebookresearch/BLINK/blob/97d4a291a5b6fcb26e92382df870ab09c87ed069/blink/main_dense.py#L272

ledw avatar Jan 27 '21 04:01 ledw