Install RAPIDS with GPU support

Overview

Teaching: 5 min min
Exercises: 0 min

Questions

• How to install Rapids with GPU support to Palmetto

Objectives

Installing RAPIDS can be a little bit tricky. So here is what I have done so far at the time of writing this note.

Request Tesla V100 node:

$ qsub -I -l select=1:ncpus=16:mpiprocs=16:mem=62gb:ngpus=2:gpu_model=v100,walltime=72:00:00

Load necessary modules:

$ module load anaconda3/2019.10-gcc/8.3.1  cuda/10.2.89-gcc/8.3.1

Follow the tutorial here: https://rapids.ai/start.html

Here is the command to create conda environment and install rapids that is successful for me:

$ conda create -n rapids-0.17 -c rapidsai -c nvidia -c conda-forge \
    -c defaults rapids-blazing=0.17 python=3.7

Key Points

• Rapids, gpu