Skip to content

Instantly share code, notes, and snippets.

@kayush2O6
Last active July 20, 2022 14:18
Show Gist options
  • Save kayush2O6/a0d2f389bc013c3e2345c268034ce78d to your computer and use it in GitHub Desktop.
Save kayush2O6/a0d2f389bc013c3e2345c268034ce78d to your computer and use it in GitHub Desktop.
rapids'cudf on google colab
Steps 1: Just to verify that you have all requirement satisfied, needed by rapids.
* check the gpu card (>=Pascal arch)
!nvidia-smi
* check cuda version installed (>=9.2)
!nvcc -V
*check the python and pip version (python==3.6)
!python -V; pip -V
Step 2: Install the cudf and make some adjustment in order to make it work.
!pip install cudf-cuda100==0.6
# If you try to import cudf at this point, it will give you, librmm.so not found error.#
*find the location of librmm.so using:
!find / -name librmm.so*
# It should give something like this: /usr/local/lib/python3.6/dist-packages/librmm.so #
*copy this librmm.so to your current working directory.
!cp /usr/local/lib/python3.6/dist-packages/librmm.so .
# at this point you will be able to import cudf#
Step 3: Here you have to set the path for NVVM. copy and paste the below lines to notebook cell.
import os
os.environ['NUMBAPRO_NVVM']='/usr/local/cuda-10.0/nvvm/lib64/libnvvm.so'
os.environ['NUMBAPRO_LIBDEVICE']='/usr/local/cuda-10.0/nvvm/libdevice'
Finally, you can run your code.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment