Skip to content

GPU Support #43

@xuesongwang

Description

@xuesongwang

Description

Hi there,

I've noticed that the backend device is "cpu"
from hyperimpute.plugins.core.device import DEVICE
I was wondering if there could be a way to specify the DEVICE when GPU is available?

Cheers.

Are you interested in working on this improvement yourself?

  • Yes, I am.

Metadata

Metadata

Assignees

No one assigned

    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