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