-
Notifications
You must be signed in to change notification settings - Fork 25
Open
Labels
Waiting on AuthorWaiting on AuthorWaiting on AuthorquestionFurther information is requestedFurther information is requested
Description
!nvidia-smi
This get the RAPIDS-Colab install files and test check your GPU. Run this and the next cell only.
Please read the output of this cell. If your Colab Instance is not RAPIDS compatible, it will warn you and give you remediation steps.
!pip install pynvml
!pip install bertopic
!git clone https://github.com/rapidsai/rapidsai-csp-utils.git
!python rapidsai-csp-utils/colab/env-check.py
This will update the Colab environment and restart the kernel. Don't run the next cell until you see the session crash.
!bash rapidsai-csp-utils/colab/update_gcc.sh
import os
os._exit(00)
This will install CondaColab. This will restart your kernel one last time. Run this cell by itself and only run the next cell once you see the session crash.
import condacolab
condacolab.install()
you can now run the rest of the cells as normal
import condacolab
condacolab.check()
Installing RAPIDS is now 'python rapidsai-csp-utils/colab/install_rapids.py '
The options are 'stable' and 'nightly'. Leaving it blank or adding any other words will default to stable.
!python rapidsai-csp-utils/colab/install_rapids.py stable
import os
os.environ['NUMBAPRO_NVVM'] = '/usr/local/cuda/nvvm/lib64/libnvvm.so'
os.environ['NUMBAPRO_LIBDEVICE'] = '/usr/local/cuda/nvvm/libdevice/'
os.environ['CONDA_PREFIX'] = '/usr/local'
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
Waiting on AuthorWaiting on AuthorWaiting on AuthorquestionFurther information is requestedFurther information is requested