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
14.06k stars 1.82k 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. #5708

Open yyh7773 opened 1 year ago

yyh7773 commented 1 year ago

''Hello,NAS'':GPU can't be used but i set all: and if i use CPU,there is a error :策略执行失败

Environment:

Configuration:

model_space = MyModelSpace()

上述是我的设置情况,请问应该如何解决?

yyh7773 commented 1 year ago

MyModelSpace( (conv1): Conv2d(1, 32, kernel_size=(3, 3), stride=(1, 1)) (conv2): LayerChoice( label='conv2' (0): Conv2d(32, 64, kernel_size=(3, 3), stride=(1, 1)) (1): DepthwiseSeparableConv( (depthwise): Conv2d(32, 32, kernel_size=(3, 3), stride=(1, 1), groups=32) (pointwise): Conv2d(32, 64, kernel_size=(1, 1), stride=(1, 1)) ) ) (dropout1): MutableDropout(p=Categorical([0.25, 0.5, 0.75], label='dropout')) (dropout2): Dropout(p=0.5, inplace=False) (fc1): MutableLinear(in_features=9216, out_features=Categorical([64, 128, 256], label='feature')) (fc2): MutableLinear(in_features=Categorical([64, 128, 256], label='feature'), out_features=10) ) 我的搜索空间与Demo中一样

XuanmaiXue commented 10 months ago

I got same problem, have you solved it ?