Machine Learning Guide MLG 024 Tech Stack
Use Nvidia GPUs for TensorFlow
- Choose an Nvidia GPU for deep learning due to TensorFlow's CUDA support.
- Avoid AMD GPUs because TensorFlow does not support them as effectively as Nvidia cards.
Switched from Mac to Custom PC
- Tyler Ronelli switched from MacBook Pro to a custom PC with Nvidia GPU for machine learning.
- He values the superior GPU support and performance on Ubuntu Linux over his previous Mac setup.
Build Custom PC with Ubuntu
- Build a custom desktop PC with an Nvidia 1080 TI GPU for best machine learning performance.
- Use Ubuntu Linux as the operating system for optimal TensorFlow support and stability.
Try a walking desk to stay healthy while you study or work!
Notes and resources at ocdevel.com/mlg/24
HardwareDesktop if you're stationary, as you'll get the best performance bang-for-buck and improved longevity; laptop if you're mobile.
Desktops. Build your own PC, better value than pre-built. See PC Part Picker, make sure to use an Nvidia graphics card. Generally shoot for 2nd-best of CPUs/GPUs. Eg, RTX 4070 currently (2024-01); better value-to-price than 4080+.
For laptops, see this post (updated).
OS / SoftwareUse Linux (I prefer Ubuntu), or Windows, WSL2, and Docker. See mla/12 for details.
Programming Tech StackDeep-learning frameworks. You'll use both TF & PT eventually, so don't get hung up. mlg/9 for details.
- Tensorflow (and/or Keras)
- PyTorch (and/or Lightning)
Shallow-learning / utilities: ScikitLearn, Pandas, Numpy
Cloud-hosting: AWS / GCP / Azure. mla/13 for details.
Episode SummaryThe episode discusses setting up a tech stack tailored for machine learning, emphasizing the necessity of choosing a primary programming language and framework, which, in this case, are Python and TensorFlow. The decision is supported by the ongoing popularity and community support for these tools. This preference is further influenced by the necessity for GPU optimization, which TensorFlow provides, allowing for enhanced performance through utilizing Nvidia's CUDA technology.
A notable change in the landscape is the decline of certain deep learning frameworks such as Theano, and the rise of competitors like PyTorch, which is gaining traction due to its ease of use in comparison to TensorFlow. The author emphasizes the importance of selecting frameworks with robust community support and resources, highlighting TensorFlow's lead in the market in this respect.
For hardware, the suggestion is a custom-built PC with a powerful Nvidia GPU, such as the 1080 TI, running Ubuntu Linux for best compatibility. However, for those who favor cloud services, Amazon Web Services (AWS) and Google Cloud Platform (GCP) are viable options, with a preference for GCP due to cost and performance benefits, particularly with the upcoming Tensor Processing Units (TPUs).
On the software side, the use of Pandas for data manipulation, NumPy for mathematical operations, and Scikit-Learn for shallow learning tasks provides a comprehensive toolkit for machine learning development. Additionally, the use of abstraction libraries such as Keras for simplifying TensorFlow syntax and TensorForce for reinforcement learning are recommended.
The episode further explores system architectures, suggesting a separation of concerns between a web app server and a machine learning (job) server. Communication between these components can be efficiently managed using a message queuing system like RabbitMQ, with Celery as a potential abstraction layer.
To support developers in implementing their machine learning pipelines, the recommendation extends to leveraging existing datasets, using Scikit-Learn for convenient access, and standardizing data for effective training results. The author points to several books and resources to assist in understanding and applying these technologies effectively, ending with your own workstation recommendations and building TensorFlow from source for performance gains as a potential advanced optimization step.