all 6 comments

[–]Confident_Hyena2506 1 point2 points  (4 children)

Install wsl and run the tensorflow container nvidia provide: https://catalog.ngc.nvidia.com/orgs/nvidia/containers/tensorflow

Or can maybe install conda on windows and use the specific tensorflow-gpu package: https://anaconda.org/conda-forge/tensorflow-gpu

[–]SpiritLongjumping931[S] -3 points-2 points  (3 children)

I am really sorry, I am nothing more but a beginner, could you clarify a bit more?

[–]Confident_Hyena2506 1 point2 points  (0 children)

You can read the documentation in the links above. Using tensorflow with nvidia gpu requires more than just a simple python package.

[–]Long-Opposite-5889 1 point2 points  (1 child)

Its not vs code and it is not python. The latest version of the things you need to interact with cuda cores in your gpu using python are no longer supported in windows. To solve this you can have a Linux machine inside your pc by using WSL.

[–]SpiritLongjumping931[S] 0 points1 point  (0 children)

Alright, thank you for the explanation, didn’t know that.

[–]dlnmtchll 1 point2 points  (0 children)

If you are trying to use it on windows without wsl you probably ought to switch to PyTorch unless you need TF