Bohamiann not using GPU

automl / RoBO

RoBO: a Robust Bayesian Optimization framework

BSD 3-Clause "New" or "Revised" License

482 stars 133 forks source link

Open samuelkim314 opened 2 years ago

samuelkim314 commented 2 years ago

Is there a flag or something I need to set? I have Torch installed and it is able to detect the GPU just fine.