DEV Community

Cover image for GPU config for ML
Rohith ND
Rohith ND

Posted on

GPU config for ML

I am involved in Deep Learning projects. My model takes too long to run because it uses CPU by default. I can't figure out how to set up my GPU for tensorflow and Pytorch. Can anyone assist me?

Top comments (0)