What is the best environment for deep learning on a local GPU workstation?

Hi everyone,

I recently built a local workstation for deep learning research, and I’d like to fully utilize my GPU with TensorFlow instead of using Google Colab.

Specs

  • CPU: Intel i9-11900K (8 cores, 11th Gen)
  • GPU: NVIDIA RTX 5070 (12 GB VRAM)
  • RAM: 128 GB DDR5
  • SSD: 2 TB NVMe
  • OS: Windows 11 / Ubuntu WSL2 (optional)

Goal
To train deep learning models (CNNs, Transformers) with TensorFlow using full GPU acceleration.

Tools I’m considering

  • JupyterLab
  • VS Code
  • (others welcome)

Question
Which environment is most suitable for TensorFlow GPU development on a local workstation?

(I’m a Clinical Research Coordinator, and my professor asked me to check this setup, so I’m trying to find the best environment before we start our experiments.)

Thanks in advance for any insights or setup recommendations!