Sign In Start Free Trial
Account

Add to playlist

Create a Playlist

Modal Close icon
You need to login to use this feature.
  • Book Overview & Buying GPU-Accelerated Computing with Python 3 and CUDA
  • Table Of Contents Toc
GPU-Accelerated Computing with Python 3 and CUDA

GPU-Accelerated Computing with Python 3 and CUDA

By : Niels Cautaerts, Hossein Ghorbanfekr
close
close
GPU-Accelerated Computing with Python 3 and CUDA

GPU-Accelerated Computing with Python 3 and CUDA

By: Niels Cautaerts, Hossein Ghorbanfekr

Overview of this book

Writing high-performance Python code doesn’t have to mean switching to C++. This book shows you how to accelerate Python applications using NVIDIA’s CUDA platform and a modern ecosystem of Python tools and libraries. Aimed at researchers, engineers, and data scientists, it offers a practical yet deep understanding of GPU programming and how to fully exploit modern GPU hardware. You’ll begin with the fundamentals of CUDA programming in Python using Numba-CUDA, learning how GPUs work and how to write, execute, and debug custom GPU kernels. Building on this foundation, the book explores memory access optimization, asynchronous execution with CUDA streams, and multi-GPU scaling using Dask-CUDA. Performance analysis and tuning are emphasized throughout, using NVIDIA Nsight profilers. You’ll also learn to use high-level GPU libraries such as JAX, CuPy, and RAPIDS to accelerate numerical Python workflows with minimal code changes. These techniques are applied to real-world examples, including PDE solvers, image processing, physical simulations, and transformer models. Written by experienced GPU practitioners, this hands-on guide emphasizes reproducible workflows using Python 3.10+, CUDA 12.3+, and tools like the Pixi package manager. By the end, you’ll have future-ready skills for building scalable GPU applications in Python.
Table of Contents (24 chapters)
close
close
1
Part 1: Fundamentals of GPU programming with CUDA in Python 3
6
Part 2: Performance Optimization and Advanced CUDA Topics
10
Part 3: Using High-Level Python Libraries for GPU Computation
14
Part 4: Real-World Example Applications
19
Part 5: Beyond This Book
23
Index

Initializing constant parameters

We will use a configuration dataclass throughout this chapter. Instead of passing a long list of parameters to each function or class, we'll define them once in a Config object and reference them wherever needed:

from dataclasses import dataclass

@dataclass
class Config:
    epochs: int = 5
    batch_size: int = 16
    embedding_dim: int = 256
    num_attention_heads: int = 4
    intermediate_dim: int = 128
    dropout_probability: float = 0.1
    vocab_size: int = 50257
    max_seq_length: int = 512
    num_hidden_layers: int = 3

We will use these parameters when implementing different LLM components. This is a clean and Pythonic way to manage constants and parameters passed to different parts of the code. We can define the configuration as follows:

cfg = Config()

Here, cfg provides the necessary parameter with minimal boilerplate, while still making it easy to override specific settings when necessary.

CONTINUE READING
83
Tech Concepts
36
Programming languages
73
Tech Tools
Icon Unlimited access to the largest independent learning library in tech of over 8,000 expert-authored tech books and videos.
Icon Innovative learning tools, including AI book assistants, code context explainers, and text-to-speech.
Icon 50+ new titles added per month and exclusive early access to books as they are being written.
GPU-Accelerated Computing with Python 3 and CUDA
notes
bookmark Notes and Bookmarks search Search in title playlist Add to playlist download Download options font-size Font size

Change the font size

margin-width Margin width

Change margin width

day-mode Day/Sepia/Night Modes

Change background colour

Close icon Search
Country selected

Close icon Your notes and bookmarks

Confirmation

Modal Close icon
claim successful

Buy this book with your credits?

Modal Close icon
Are you sure you want to buy this book with one of your credits?
Close
YES, BUY

Submit Your Feedback

Modal Close icon
Modal Close icon
Modal Close icon