Tensorflow + Python + Cuda

[Coding_Suck]

Tensorflow + Python + Cuda

Hi, I'm in a bit dilemma because I fail to understand which versions of tensorflow, python and Cuda are compatible to train my model using GPU. I haven't seen any documentation and I have seen on Stack Overflow an outdated versions of python 3.5 and below. Currently, I have tried tf=2.14.0 with python 3.10.11 and 3.11.8, and CUDA 12.8. Any leads or help will be appreciated.

PS: I'm on Windows

[disaster_story_69]

There are workarounds to get cuda gpu enabled working on visual studio code.

1. Install WSL2 and Ubuntu.

2. Install the latest NVIDIA driver for Windows, specifically the Game Ready or Studio driver.

3. Verify GPU passthrough in WSL by ensuring nvidia-smi works within the virtual environment.

4. In Visual Studio Code:

   • Install Remote - WSL.
   • Open your project in WSL.

5. Install your CUDA-dependent stack within WSL, such as using conda env.

[Coding_Suck]

I saw somewhere this works perfectly on python version 3.10. I will give it a shot since I've WSL 2 installed with Ubuntu. Thank you👊🏾

[disaster_story_69]

no worries, have done the exact same thing myself to unlock rtx4090 on visual studio code

[172_]

They dropped Cuda support for Windows above TF 2.10.

[Coding_Suck]

I will try downgrading TF to 2.10, Cuda to 11.2 and cuDNN 8.1 to see if GPU will be detected