Machine learning models on a GPU
Saturn Cloud has a built in GPU image for RAPIDS that has all the required libraries to get started using RAPIDS. When creating a new resource, select the
saturn-rapids image. Once the resource starts, your RAPIDS code should be ready to run. This will also work well with Dask, and is how the Saturn Cloud RAPIDS examples run.
If you want to create your own image, you will need to install the correct version of RAPIDS based on your CUDA version. You can get this from the
rapidsai conda channel.
channels: - rapidsai - defaults dependencies: - rapids=0.14.1=cuda10.1_py37_0