Tensorflow with GPU on Windows WSL using Docker

In this video we show you how to run Tensorflow with GPU on Windows using WSL (WSL2) and Docker. There are several steps that should be completed in order. However, the initial challenges are worth is. This is the best way to locally run machine learning tasks on windows.

Companion Article:

If this video helped you out, be sure to like and subscribe for more content!

Links (in order of installation):

Master Tensorflow and Keras with the creator of Keras, François Chollet! Plus, you can help support KNuggies with this affiliate link 🤑:

Join this channel to get access to perks:

Follow on Social Media to keep up with the latest KNuggies posts:

Join the Discord server for anouncements:

All KNuggies Medium articles can be found here:

[KNuggies]

Master Tensorflow and Keras with the creator of Keras, François Chollet! Plus, you can help support KNuggies with this affiliate link 🤑:

[bricef]

Amazing tutorial. Almost 1 year later, this works seamlessly.

I've been trying for over 8 hours today to get Tensorflow GPU working on Windows 11 to no avail..
Watching this, got it set up in 15 minutes.

Can't thank you enough, wish I would have found this sooner... Cheers!!

[Sebastian-hvjz]

I used this tutorial today, Jan 2023. It still works and it is the ONLY tutorial about installation of TF on Windows with NVIDIA on YT that works! Good job! I also just got Google Coral Dev board to play with.

[brunaguterres]

Best tutorial on the YouTube for using docker woth GPU support in windows through WSL. It worked just fine in January 2023

[thivuxhale]

one thing to note about the prerequisites is that windows has to be in version 21H2, if you get version 21H1, you can't access GPU inside WSL. took me 2 days to figure out :v

[adrianfletcher]

I love running into these tutorials that have just the right amount of rigor to get you started.

[nikhielsingh]

best video on YouTube, works in August 2023

[galrozental]

Absolutely amazing tutorial.
I got so fed up with trying to make the latest version of tensorflow work with gpu that I almost just installed an old version that worked for me before.
But this is SO MUCH BETTER!
so easy to set up, works flawlessly, 10/10.
I only need to figure out my workflow with git and vscode to make it as easy as possible but hopefully that is the simple part.
Thanks you so much!

[datapro]

Your tutorial was most valuable, as is the included repository. Thank you!

[brunaguterres]

It is for sure the best video regarding docker usage on Windows and VS code. Congrats!

[genericwannabe]

Just fyi, you could use CuPy instead of Numpy if you want to make use of your GPU for normal Numpy related things. It is more or less swappable for most Numpy commands, so it’s easy to modify code between the two.

[maloman]

Really good tutorial, all you need to start in less than seventeen minutes! Thanks!

[DexkillTutorials]

Hey everyone I could get rid of the errors after installing CUDA on the WLS2 instance, thank you for the tutroial!

[aliriano]

Really good tutorial. Helped me figure out how to get the GPU running!

[nikitasmirnov]

If you installed on wsl distributive such as ubuntu 20.04, you can face unexpected errors, when you will try open "new wsl window". I solved this problem by choosing an option: "New WSL Window using Distro" and then just selected my installed system from the dropdown

[totoflex...]

I had to struggle a bit for some unexpected error messages but once it was fixed, it worked very well ! Thanks :)

[profiorucci]

docker run -it --rm -p 8888:8888 --gpus all

[stevebailey]

Excellent - that told me exactly what I need to know - well done!

[pramanikd]

Fantastic tutorial! Everything was to the point and easy to follow.

[CompleteReality]

Great tutorial! Please keep teaching us all the good stuff ;)