Add option to install cpu-only PyTorch dependency #60

Open

Labels

opened

on Feb 14, 2025

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).

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

Metadata

Assignees

No one assigned

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests