-
Notifications
You must be signed in to change notification settings - Fork 1.3k
Closed
Labels
enhancementNew feature or requestNew feature or requestfeature requestRequest for a new feature to be added to AccelerateRequest for a new feature to be added to Accelerate
Description
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 requestNew feature or requestfeature requestRequest for a new feature to be added to AccelerateRequest for a new feature to be added to Accelerate