DEV Community

Cover image for GPU config for ML
Rohith ND
Rohith ND

Posted on

2

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)

AWS Security LIVE!

Join us for AWS Security LIVE!

Discover the future of cloud security. Tune in live for trends, tips, and solutions from AWS and AWS Partners.

Learn More

👋 Kindness is contagious

Please leave a ❤️ or a friendly comment on this post if you found it helpful!

Okay