Skip to content

Feature request: Specifying GPU IDs #543

@tmabraham

Description

@tmabraham

It would be ideal to specify the GPU IDs that a script can use. i typically like to have a GPU ID argument in my script so that I can set the device, but if Accelerate is handling the devices I can't specify. The current alternative is to use CUDA_VISIBLE_DEVICES but a dedicated argument in accelerate config or in the Accelerator object would be ideal.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or requestfeature requestRequest for a new feature to be added to Accelerate

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions