Use CUDA_VISIBLE_DEVICES with accelarator

debwol · August 18, 2021, 1:41pm

I try to set CUDA_VISIBLE_DEVICES on the command line. But when accelerator set the device, it chose the wrong one. What should I do?
Thanks

sgugger · August 30, 2021, 6:16pm

How do you know it chose the wrong one? What command did you type exactly?

Topic		Replies	Views
How to use specific gpu in accelerate? 🤗Accelerate	10	5412	April 25, 2024
Accelarator can't detect my GPUs? 🤗Accelerate	10	873	March 29, 2024
Unable to use multiple GPUs with CUDA_VISIBLE_DEVICES Beginners	0	98	June 17, 2024
Can't set attribute 'device', for some reason i need to train model on only one gpu on a mutlti gpu machine Beginners	2	327	September 16, 2024
How to use specified GPUs with Accelerator to train the model? Beginners	15	25805	August 23, 2024