Last active
July 20, 2022 14:18
-
-
Save kayush2O6/a0d2f389bc013c3e2345c268034ce78d to your computer and use it in GitHub Desktop.
rapids'cudf on google colab
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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