Skip to Content
40 Algorithms Every Programmer Should Know
book

40 Algorithms Every Programmer Should Know

by Imran Ahmad
June 2020
Intermediate to advanced
382 pages
11h 39m
English
Packt Publishing
Content preview from 40 Algorithms Every Programmer Should Know

Using GPUs for data processing in Python

GPUs are great for data processing in a multidimensional data structure. These data structures are inherently parallelizable. Let's see how we can use the GPU for multidimensional data processing in Python:

  1. First, let's import the Python packages that are needed:

import numpy as npimport cupy as cpimport time
  1. We will be using a multidimensional array in NumPy, which is a traditional Python package that uses the CPU.

  1. Then, we create a multidimensional array using a CuPy array, which uses the GPU. Then, we will compare the timings:

### Running at CPU using Numpystart_time = time.time()myvar_cpu = np.ones((800,800,800))end_time = time.time()print(end_time - start_time)### Running at GPU using CuPy ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

50 Algorithms Every Programmer Should Know - Second Edition

50 Algorithms Every Programmer Should Know - Second Edition

Imran Ahmad
Grokking Algorithms

Grokking Algorithms

Aditya Bhargava

Publisher Resources

ISBN: 9781789801217Supplemental Content