Cupy介绍与安装

Cupy概述

可以把cupy看做numpy的GPU版本，使用GPU强大的并行计算能力。numpy是很重要的基础设施库，许多库都是基于它构建的，比如：Scipy,Pandas,Matplotlib,Scikit-learn,Statsmodels,SymPy等。由此可见numpy的重要性，而cupy作为其GPU版本，对于需要使用GPU加速计算的项目更加重要。

Cupy安装

可以直接使用pip install cupy安装，但是因为cupy这个库底层依赖的复杂性，这种方法很大概率会遇到各种各样的报错，比如C++编译报错等等。

因此，适合使用预编译的二进制包进行安装，命令如下：
对于CUDA版本为11.x的，使用：pip install cupy-cuda11x
对于CUDA版本为12.x的，使用：pip install cupy-cuda12x
注意不要把x改为具体的数字，比如我的CUDA版本为12.1，安装命令是pip install cupy-cuda12x，而不是错误的命令pip install cupy-cuda121

还有，确保已经提前安装了CUDA和cuDNN。