cupy 会自动使用 cuda 和 gpu 吗?

Does cupy automatically use cuda and the gpu?

我知道对于其他允许 python 与 GPU 一起使用的库,您必须指定您使用的是 cuda,否则这些函数将工作但不使用 cuda,例如 numba。 cupy也是这样,还是自动使用GPU?

是的,CuPy 自动使用 CUDA

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

pip install cupy-cudaXX

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