How To Install python-pycuda on Ubuntu 18.04

In this tutorial we learn how to install python-pycuda on Ubuntu 18.04. python-pycuda is Python module to access Nvidia‘s CUDA parallel computation API

Introduction

In this tutorial we learn how to install python-pycuda on Ubuntu 18.04.

What is python-pycuda

python-pycuda is:

PyCUDA lets you access Nvidia‘s CUDA parallel computation API from Python. Several wrappers of the CUDA API already exist–so what’s so special about PyCUDA?

  • Object cleanup tied to lifetime of objects. This idiom, often called RAII in C++, makes it much easier to write correct, leak- and crash-free code. PyCUDA knows about dependencies, too, so (for example) it won’t detach from a context before all memory allocated in it is also freed.
  • Convenience. Abstractions like pycuda.driver.SourceModule and pycuda.gpuarray.GPUArray make CUDA programming even more convenient than with Nvidia’s C-based runtime.
  • Completeness. PyCUDA puts the full power of CUDA’s driver API at your disposal, if you wish.
  • Automatic Error Checking. All CUDA errors are automatically translated into Python exceptions.
  • Speed. PyCUDA’s base layer is written in C++, so all the niceties above are virtually free.
  • Helpful Documentation.

There are three methods to install python-pycuda on Ubuntu 18.04. We can use apt-get, apt and aptitude. In the following sections we will describe each method. You can choose one of them.

Install python-pycuda Using apt-get

Update apt database with apt-get using the following command.

sudo apt-get update

After updating apt database, We can install python-pycuda using apt-get by running the following command:

sudo apt-get -y install python-pycuda

Install python-pycuda Using apt

Update apt database with apt using the following command.

sudo apt update

After updating apt database, We can install python-pycuda using apt by running the following command:

sudo apt -y install python-pycuda

Install python-pycuda Using aptitude

If you want to follow this method, you might need to install aptitude first since aptitude is usually not installed by default on Ubuntu. Update apt database with aptitude using the following command.

sudo aptitude update

After updating apt database, We can install python-pycuda using aptitude by running the following command:

sudo aptitude -y install python-pycuda

How To Uninstall python-pycuda on Ubuntu 18.04

To uninstall only the python-pycuda package we can use the following command:

sudo apt-get remove python-pycuda

Uninstall python-pycuda And Its Dependencies

To uninstall python-pycuda and its dependencies that are no longer needed by Ubuntu 18.04, we can use the command below:

sudo apt-get -y autoremove python-pycuda

Remove python-pycuda Configurations and Data

To remove python-pycuda configuration and data from Ubuntu 18.04 we can use the following command:

sudo apt-get -y purge python-pycuda

Remove python-pycuda configuration, data, and all of its dependencies

We can use the following command to remove python-pycuda configurations, data and all of its dependencies, we can use the following command:

sudo apt-get -y autoremove --purge python-pycuda

References

Summary

In this tutorial we learn how to install python-pycuda package on Ubuntu 18.04 using different package management tools: apt, apt-get and aptitude.