Setting up Docker for Machine Learning
The Dockerfile I use to set up my machine learning environment.
This post is mainly meant for coworkers, but it might be useful for others as well. I'll be sharing the Dockerfile that I use to set up my machine learning environment. It's based on the
huggingface/transformers-pytorch-gpu image, but I've added a few things (Infiniband support, upgraded Pip packages, a utility to show GPU status, etc.) to make it a bit more useful for me.
Place all of the below files in a new directory. Feel free to add or remove anything you want — I'll likely update this post as I make changes to my setup.
After you've created the files, you can build the image with:
If you liked the article, don't forget to share it and follow me at @nebrelbug on Twitter.