-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Closed
Labels
fabriclightning.fabric.Fabriclightning.fabric.FabricfeatureIs an improvement or enhancementIs an improvement or enhancementhelp wantedOpen to be worked onOpen to be worked onver: 2.4.x
Description
Bug description
when run this codes
fabric run ./path/to/train.py \
--devices=auto \
--accelerator=auto
get problem below
Error: Invalid value for '--accelerator': 'auto' is not one of 'cpu', 'gpu', 'cuda', 'mps', 'tpu'.
What version are you seeing the problem on?
v2.4
How to reproduce the bug
No response
Error messages and logs
# Error messages and logs here please
Environment
Current environment
#- PyTorch Lightning Version (e.g., 2.4.0):
#- PyTorch Version (e.g., 2.4):
#- Python version (e.g., 3.12):
#- OS (e.g., Linux):
#- CUDA/cuDNN version:
#- GPU models and configuration:
#- How you installed Lightning(`conda`, `pip`, source):
More info
No response
Metadata
Metadata
Assignees
Labels
fabriclightning.fabric.Fabriclightning.fabric.FabricfeatureIs an improvement or enhancementIs an improvement or enhancementhelp wantedOpen to be worked onOpen to be worked onver: 2.4.x
Activity
lantiga commentedon Dec 5, 2024
hey @wkd88 that's correct, would you want to give it a shot in a quick PR?
here's the file that needs to be updated: https://github.com/Lightning-AI/pytorch-lightning/blob/master/src/lightning/fabric/cli.py
GeminiLn commentedon Dec 7, 2024
Hi @lantiga I want to try a PR on this issue.
I’ve updated the file to include the "auto" option for the accelerator. You can find the PR here: #20473 .
fnhirwa commentedon Jun 10, 2025
Hello @Borda I would like to take on this issue.
Thanks
devices
andaccelerator
as cli arguments #20913