BLINK
BLINK copied to clipboard
raise AssertionError("Torch not compiled with CUDA enabled") on Mac
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 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