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。
我知道对于其他允许 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。