【Question】How to use GPU training, just install tensorflow-gpu？ will there be better performance if using a larger model?

google / ml-compiler-opt

Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.

Apache License 2.0

629 stars 93 forks source link

【Question】How to use GPU training, just install tensorflow-gpu？ will there be better performance if using a larger model? #318

Closed 18liumin closed 11 months ago

mtrofin commented 11 months ago

Yes, just follow the tensorflow guidance on how to use GPU acceleration. Training time may benefit indeed. It won't impact inference, because the generated code (AOT compiled model) does not use GPU.