May 15, 2023, 6:21 p.m. | Sascha Kirch

Towards Data Science - Medium towardsdatascience.com

Multi-GPU Training on a single GPU System in 3 Minutes

An Advanced Guide for TensorFlow

Photo by Chris Liverani on Unsplash

I want to share with you a neat little trick on how I test my multi GPU training code on a single GPU.

Motivation

I guess the problem is obvious and you probably experienced it yourself. You want to train a deep learning model and you want to take advantage of multiple GPUs, a TPU or even multiple workers …

deep learning distributed systems gpu python tensorflow

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

C003549 Data Analyst (NS) - MON 13 May

@ EMW, Inc. | Braine-l'Alleud, Wallonia, Belgium

Marketing Decision Scientist

@ Meta | Menlo Park, CA | New York City