Skip to main content

Using Rapids cuDF in a Colab notebook

During last Spark+AI Summit Europe 2019 I had a chance to attend a talk from Miguel Martinez who was presenting Rapids, the new Open Source framework from NVIDIA for GPU accelerated end-to-end Data Science and Analytics.

Fig. 1 - Overview of the Rapids eco-system

Rapids is a suite of Open Source libraries:

  • cuDF
  • cuML
  • cuGraph
  • cuXFilter

I enjoied the presentation and liked the idea of this initiative, so I wanted to start playing with the Rapids libraries in Python on Colab, starting from cuDF, but the first attempt came with an issue that I eventually solved. So in this post I am going to share how I fixed it, with the hope it would be useful to someone else running into the same blocker.
I am assuming here you are already familiar with Google Colab. I am using Python 3.x as Python 2 isn't supported by Rapids.
Once you have created a new notebook in Colab, you need to check if the runtime for it is set to use Python 3 and uses a GPU as hardware accelerator. You can check and eventually set it from the toolbar menu Runtime -> Change runtime type. Then you have to check that Colab has given you access to a NVIDIA Tesla GPU, as it is supported by Rapids. You can do this check from the notebook code by running


and look for the GPU type, as highligthed in figure 2:

Fig. 2 - The output of the nvidia-smi command in a Colab notebook

The next step would be the installation of the cuDF library for CUDA 10.x. You can use pip from the notebook:

!pip install cudf-cuda100

At the end of the installation, before importing the cuDF package, you need to do an extra setup that I am going to explain below. If you don't do it, any time you try to execute the cuDF API, you will receive the following misleading error:

Fig. 3 - The error message in Colab when trying to use the cuDF API

The message states that the cudatoolkit isn't installed in your environment, so the numba package cannot be find, which isn't true, as cudatoolkit is already part of the Colab environment. The real root cause of this error is that the paths to the drivers are different in the runtime from what expected by the library. So you have to locate them first

dev_lib_path = !find / -iname 'libdevice'
nvvm_lib_path = !find / -iname ''

and setup the environment accordingly

import os

if len(dev_lib_path) > 0:
    os.environ['NUMBAPRO_LIBDEVICE'] = dev_lib_path[0]
    print('The device lib is missing.')
if len(nvvm_lib_path) > 0:
    os.environ['NUMBAPRO_NVVM'] = nvvm_lib_path[0]
    print('NVVM is missing.')

I have also put extra checks in case the paths shouldn't be found, which would probably be useless, but you never know :)

Now you can import cuDF, numpy and any other Python package you would need

import cudf
import numpy as np

and start playing with GPU Data Frames

df = cudf.DataFrame()

Enjoy it!

I will share my impressions on this library and the others in Rapids as soon as I have completed some PoCs.


  1. Feedback from Miguel Martinez (NVIDIA): by installing cuDF using pip, you will get the release 0.61. In order to install a latter or the latest release you have to use conda onstead. Thanks Miguel for the heads up.

  2. This video helps me to understand Matplotlib whats your opinion guys.

  3. Nice Post.. really helpful to clear my small concepts. thanks :)Keep doing more, waiting to read your next blog security testing services. Visit here for more information about Vulnerability assessment services and penetration testing services.

  4. Nice post. You pointed on very important facts by this post. Oracle EBS Test Automation Tools

  5. This Content is simply amazing, and helpful for students and business people. Gathered lots of information and waiting to see more updates.
    RPA Course in Chennai
    RPA Online Course
    RPA Training in Coimbatore

  6. This article is a creative one and the concept is good to enhance our knowledge. Waiting for more updates.
    Artificial Intelligence Online Training
    Artificial Intelligence Online Course

  7. Nice blog, very informative content.Thanks for sharing, waiting for the next update…
    Why learn Python?
    Why is Python so popular?

  8. Pretty Post! Thank you so much for sharing this good content, it was so nice to read and useful to improve my knowledge as an updated one, keep blogging.

    Python Certification Training in Electronic City

  9. I like reading the above article because it clearly explains everything and is both entertaining and effective. Thank you for your time and consideration, and best of luck with your future articles.

    Data Engineering Solutions 

    Artificial Intelligence Services

    Data Analytics Services

    Data Modernization Services


Post a Comment

Popular posts from this blog

Exporting InfluxDB data to a CVS file

Sometimes you would need to export a sample of the data from an InfluxDB table to a CSV file (for example to allow a data scientist to do some offline analysis using a tool like Jupyter, Zeppelin or Spark Notebook). It is possible to perform this operation through the influx command line client. This is the general syntax: sudo /usr/bin/influx -database '<database_name>' -host '<hostname>' -username '<username>'  -password '<password>' -execute 'select_statement' -format '<format>' > <file_path>/<file_name>.csv where the format could be csv , json or column . Example: sudo /usr/bin/influx -database 'telegraf' -host 'localhost' -username 'admin'  -password '123456789' -execute 'select * from mem' -format 'csv' > /home/googlielmo/influxdb-export/mem-export.csv

TagUI: an Excellent Open Source Option for RPA - Introduction

 Photo by Dinu J Nair on Unsplash Today I want to introduce  TagUI , an RPA (Robotic Process Automation) Open Source tool I am using to automate test scenarios for web applications. It is developed and maintained by the AI Singapore national programme. It allows writing flows to automate repetitive tasks, such as regression testing of web applications. Flows are written in natural language : English and other 20 languages are currently supported. Works on Windows, Linux and macOS. The TagUI official documentation can be found  here . The tool doesn't require installation: just go the official GitHub repository and download the archive for your specific OS (ZIP for Windows, tar.gz for Linux or macOS). After the download is completed, unpack its content in the local hard drive. The executable to use is named  tagui  (.cmd in Windows, .sh for other OS) and it is located into the  <destination_folder>/tagui/src  directory. In order to use it, the Google Chrome web browser ne