June 2020
Intermediate to advanced
382 pages
11h 39m
English
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:
First, let's import the Python packages that are needed:
import numpy as npimport cupy as cpimport time
We will be using a multidimensional array in NumPy, which is a traditional Python package that uses the CPU.
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 ...