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 (1)

Collapse
 
mitul_garg_680a8c71037829 profile image
Mitul Garg

env-doctor detects if TensorFlow can use your GPU and explains why it can't.

It knows TensorFlow dropped native Windows GPU support after 2.10 and
recommends WSL2 setup or correct Linux installation commands.

Run: env-doctor check
GitHub: github.com/mitulgarg/env-doctor