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.