Skip to content

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

Description

@cwloka

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.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions