How to Set Your Windows Workstation Up for Deep Learning

Install WSL2, CUDA, PyTorch, Jupyter, and other essential tools

My articles are free to read for everyone. If you don’t have a Medium subscription, read this article by following this link.

Photo by JC Gellidon on Unsplash

Honestly, we’ve all been there at some point - we bought a new workstation with a shiny new Nvidia RTX 4080 GPU to work on “Deep Learning” projects, dual-booted it with Windows and Ubuntu, logged into Windows, installed some games, the Adobe suite and other production software, and hardly (if ever) logged back into Ubuntu.

I won’t necessarily blame this on you lacking the motivation, but rather on the barrier to entry you’re facing. Trust me when I say I know how cumbersome it is to have to constantly switch between, update and maintain two operating systems at once. Ubuntu doesn’t support gaming, designing, and other production software suites that Windows supports, however, the development environment and control provided by Ubuntu is second to none. Wouldn’t it be amazing if you can do all of your development, gaming, video production and design in just one OS itself?

Well, this is where the Windows Subsystem for Linux 2 (WSL2) comes in. Don’t get me wrong, you can absolutely setup almost any environment in Windows itself with the plethora of…