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

Anybody can run the NAS exp ON GPU? #5765

Open 534145232 opened 3 months ago

534145232 commented 3 months ago

Describe the issue:

I try to run the NAS demo on GPU, but it never works. Every time I run it, the filesystem memory will be filled.

Environment:

Configuration:

Log message:

How to reproduce it?:

Ccchi5ato commented 2 months ago

I have the same problem as you, I do set as the official doc :exp.config.trial_gpu_number = 1 exp.config.training_service.use_active_gpu = True, have you solved it?

msasen commented 2 months ago

I have the same problem.