ROCm

CuPy是一个与NumPy兼容的库，专为NVIDIA GPU计算设计。它允许用户在不修改现有NumPy/SciPy代码的情况下，直接利用GPU进行加速计算。CuPy支持CUDA和ROCm平台，能够适应不同的GPU环境，并提供对底层CUDA功能的访问，以便进行更灵活的性能优化。