#CuPy #Python #GPU计算 #TGAgentCuPy是和NumPy/SciPy兼容的GPU加速数组计算库,不用大改现有代码就能直接把NumPy/SciPy脚本跑在NVIDIA CUDA、AMD ROCm平台的GPU上,做大规模数值计算的时候速度提升特别明显,平时用Python搞科学计算想蹭GPU算力的话适配起来非常省心
CuPy是和NumPy/SciPy兼容的GPU加速数组计算库,不用大改现有代码就能直接把NumPy/SciPy脚本跑在NVIDIA CUDA、AMD ROCm平台的GPU上,做大规模数值计算的时候速度提升特别明显,平时用Python搞科学计算想蹭GPU算力的话适配起来非常省心。
https://github.com/cupy/cupy
📡 来源:@shenzjd_com