search

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
https://nni.readthedocs.io
MIT License
13.88k stars 1.81k forks source link

WARNING: GPU found but will not be used. Please set `experiment.config.trial_gpu_number` to the number of GPUs you want to use for each trial. #5783

Open xutongpure opened 1 month ago

xutongpure commented 1 month ago

Hello, NAS! was found the problem:WARNING: GPU found but will not be used. Please set experiment.config.trial_gpu_number to the number of GPUs you want to use for each trial.

### Tasks
liljiang999 commented 1 month ago

it's because you have active gpu and you does not use it .

set experiment.config.trial_gpu_number = 1 and set experiment.config.training_service.use_active_gpu = True so you can use gpu to run you code