python - Cupy 会自动使用 cuda 和 gpu 吗？

Question

我知道对于允许将 python 与 GPU 一起使用的其他库，您必须指定您正在使用 cuda，否则这些函数将起作用但不能使用 cuda，例如 numba。Cupy 是否也是如此，还是它会自动使用 GPU？

Answer 1

是的，CuPy自动使用CUDA.

在安装之前CuPy，您需要设置您的CUDA环境。然后，您可以CuPy使用特定CUDA版本进行安装，如下所示：

pip install cupy-cudaXX

其中 XX 对应于您的版本，CUDA例如cupy-cuda90. CUDA 9.0有关详细信息，请参阅https://docs-cupy.chainer.org/en/stable/install.html#install-cupy。