if cuda import, it don't mean it is cuda_capable. To check this return:

return cuda.cuda_available