Help with GPUs on NCAR Casper

I’m interested in playing with the GPUs on Casper. I have managed to log in via https://jupyterhub.ucar.edu/ and get a session with a GPU. I am using the NPL20200417 kernel, as this appears to have the latest versions of most packages. But I can’t get cupy or numba to recognize any cuda device. E.g.

from numba import cuda
cuda.select_device(0)

gives

CudaSupportError: Error at driver init: 
[100] Call to cuInit results in CUDA_ERROR_NO_DEVIC

Before spending lots of time fiddling around, I thought I would ask @jhamman or @andersy005 whether you have a working example you could share of how to do this.

Never mind! I think I figured it out. I had to actually request GPUs on login…

2 Likes