Skip to content

MPS support #152

@francescomalandrino

Description

@francescomalandrino

Is your feature request related to a problem? Please describe.

pythae supports CUDA and CPU, but no MPS

Describe the solution you'd like

I would like pythae to use MPS if no CUDA is available

Additional context

fixed (in a rudimentary fashion) by my pull request #151

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions