我知道对于允许将 python 与 GPU 一起使用的其他库，您必须指定您正在使用 cuda，否则这些函数将起作用但不能使用 cuda，例如 numba。Cupy 是否也是如此，还是它会自动使用 GPU？
问问题
518 次
1 回答
2
是的，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。
于 2019-05-15T22:46:41.230 回答