mitmul/ssai-cnn

Can it be run without cuda?

aleksandrmelnyk opened this issue · 1 comments

Am trying to start predict.py, but throws exception, RuntimeError: CUDA environment is not correctly set up
Though am not using --gpu parameter.

I think you are supposed to explicitly set --gpu -1 as opposed to using the default setting; I believe --gpu 0 instructs the cupy library that is used in the program to use the first available gpu.