Enable manual selection of the GPU #101

Open

Assignees

opened

Currently, the user selects the GPU where is the CudaImage with CudaGetMaxFlopsDev. Enabling the possibility to select the GPU manually is necessary. Similar to what torch does with torch.tensor(x, device='cuda:2').

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

Metadata

Assignees

axel-grc

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests