Cupy автоматически использует Cuda и GPU?

Question

Я знаю, что с другими библиотеками, которые позволяют использовать python с графическим процессором, вы должны указать, что вы используете cuda, иначе функции будут работать, но не будут использовать cuda, как numba. То же самое относится и к Cupy, или он автоматически использует графический процессор?

Answer 1

Да, CuPy автоматически использует CUDA.

Перед установкой CuPy необходимо настроить среду CUDA.Затем вы можете установить CuPy с определенной версией CUDA следующим образом:

pip install cupy-cudaXX

, где XX соответствует вашей CUDA версии, например cupy-cuda90 для CUDA 9.0.Подробнее см. https://docs -cupy.chainer.org / ru / stable / install.html # install-cupy .