search

KarhouTam / FL-bench

Benchmark of federated learning. Dedicated to the community. 🤗
GNU General Public License v3.0
503 stars 81 forks source link

On how to assign a gpu id to each task #120

Closed Heisenberg1159 closed 3 weeks ago

Heisenberg1159 commented 3 weeks ago

What is your question?

image

I don't know how to control the gpu id parameter in this project, I want to assign a gpu to each task.

Additional Context

No response

KarhouTam commented 3 weeks ago

Have you set the num_gpus argument in the config file?

If so, please check if your server has set CUDA_VISIBLE_DEVICES environment variable. FL-bench gives the higher priority to the environment variable setting.