SMILER icon indicating copy to clipboard operation
SMILER copied to clipboard

Add option for CUDA-based models to run CPU-only

Open cwloka opened this issue 3 years ago • 1 comments

Currently any deep learning model in SMILER expects a CUDA-compatible GPU to be available, and SMILER will fail if that is unavailable. In order to more flexibly support a wider range of operating systems and machines, it would be helpful to have the option to run models in CPU mode.

We may want to have this as a user-specified parameter, or base it off of a systems check to see if a CUDA-compatible GPU is available. Some research will likely need to be done to see which is the most feasible to implement.

cwloka avatar Feb 14 '22 03:02 cwloka

Progress update, the deep learning models SAM, BMS, and SalGan work without a gpu

richanghmc avatar Apr 20 '22 21:04 richanghmc