-
Book Overview & Buying
-
Table Of Contents
GPU-Accelerated Computing with Python 3 and CUDA
By :
To be able to run the example code provided in this chapter, you will need a system equipped with an NVIDIA GPU that has at least 10 GB of memory. We use Pixi as a dependency management tool that installs all of the required packages, including JAX and dependent libraries. To set up the environment, simply run pixi install in the root directory of the book's GitHub repository. This will ensure that all dependencies are installed. Once the installation is complete, you can activate the environment using pixi shell.
Then, run the following:
pixi run jupyter-lab
The chapter's code is available on GitHub and can be accessed via this link: https://github.com/PacktPublishing/GPU-Accelerated-Computing-with-Python-3-and-CUDA. We recommend cloning the repository and following the instructions in the README file to get started with building the environment.