Can it run on GPU? - Githubissues

google / model_search

Apache License 2.0

3.26k stars 462 forks source link

Open hw446 opened 3 years ago

hw446 commented 3 years ago

It looks like it recognizes the GPU, but it doesn't use the GPU.

davidxiaozhi commented 3 years ago

Mine is OK and you can check the runtime log, for example it needs CUDNN8's dynamic library