CuPy - GPU向けのNumPyおよびSciPy代替
(github.com/cupy)- "NumPy & SciPy for GPU"
- GPUアクセラレーテッドコンピューティング向けの、NumPy/SciPy互換配列ライブラリ
- 既存のNumPy/SciPyコードを、NVIDIA CUDAまたはAMD ROCmプラットフォーム上で実行できるように置き換え可能
- 低レベルのCUDA機能にもアクセス可能
ndarrayを既存のCUDA C/C++プログラムに渡したり、性能向上のためにStreamsを使ったり、CUDA Runtime APIを直接呼び出したりできる
まだコメントはありません。