У меня есть код python, который я пытаюсь ускорить с помощью Cuda. Я использовал функцию @jit. Как узнать, действительно ли код запускается на gpu? Есть ли способ проверить/проверить это?Выполняет ли @jit код python на gpu, если мы явно не упоминаем цель?
Вы должны использовать numba.cuda.jit для запуска вашей jitted функции на GPU. Более того, эта функция должна быть записана в виде ядра CUDA (http://numba.pydata.org/numba-doc/0.30.1/cuda/kernels.html). Когда функция запускает некоторый монитор графического процессора (например, nvidia-smi для Linux), можно использовать для просмотра нагрузки графического процессора и проверки того, что он участвует в расчете.