Sparse-Matrix_CUDA

Installation CUDA in your notebook

Check Python Version (Python 3.6.9)

!python --version

Check Ubuntu Version (Ubuntu 18.04.3 LTS)

!lsb_release -a

Check CUDA/cuDNN Version (Built on Sun_Jul_28_19:07:16_PDT_2019 / Cuda compilation tools, release 10.1, V10.1.243)

!nvcc -V && which nvcc

Check GPU

!nvidia-smi

Install RAPIDS

!git clone https://github.com/rapidsai/rapidsai-csp-utils.git
!bash rapidsai-csp-utils/colab/rapids-colab.sh stable

import sys, os

dist_package_index = sys.path.index('/usr/local/lib/python3.6/dist-packages')
sys.path = sys.path[:dist_package_index] + ['/usr/local/lib/python3.6/site-packages'] + sys.path[dist_package_index:]
sys.path
exec(open('rapidsai-csp-utils/colab/update_modules.py').read(), globals())

Try on this example

from cupy.sparse import random

S = random(100000, 2000, density=6e-2)
S.get()