nni
nni copied to clipboard
Anybody can run the NAS exp ON GPU?
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:
- NNI version: 3.0rc1
- Training service (local|remote|pai|aml|etc): local
- Client OS: windows11/ubuntu
- Server OS (for remote mode only):
- Python version: 3.8
- PyTorch/TensorFlow version: 2.0
- Is conda/virtualenv/venv used?: yes
- Is running in Docker?: no
Configuration:
- Experiment config (remember to remove secrets!):
- Search space:
Log message:
- nnimanager.log:
- dispatcher.log:
- nnictl stdout and stderr:
How to reproduce it?:
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?
I have the same problem.