Skip to content

Add option to install cpu-only PyTorch dependency #60

@ddundo

Description

@ddundo

Installing PyGEM with PyTorch+CUDA is about 7GB heavier than installing the CPU version of PyTorch. So it would be great to add an option to install PyGEM with the cpu version of it - especially for testing (#52 #53).

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions