Cupy 会自动使用 cuda 和 gpu 吗？

【问题标题】：Does cupy automatically use cuda and the gpu?Cupy 会自动使用 cuda 和 gpu 吗？
【发布时间】：2019-10-03 01:13:20
【问题描述】：

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

【问题讨论】：

标签： python cupy

【解决方案1】：

是的，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。

【讨论】：