Running CellPose with GPU

cc @pape

I activated the cellpose-gpu environment and then within a jupyter notebook I executed these lines of code from the notebook that comes with cellpose: https://github.com/MouseLand/cellpose/blob/master/notebooks/run_cellpose_GPU.ipynb

import numpy as np
import time, os, sys
from urllib.parse import urlparse
import skimage.io
import matplotlib.pyplot as plt
import matplotlib as mpl
%matplotlib inline
mpl.rcParams['figure.dpi'] = 300

from urllib.parse import urlparse
import mxnet as mx
from cellpose import utils

use_GPU = utils.use_gpu()
print('GPU activated? %d'%use_GPU)

But I got as a response:

CUDA version not installed/working, will use CPU version.
GPU activated? 0