前言 之前讀碩班常常要用 FFT 來處理光學的問題，通常都是會用 MATLAB 來撰寫程式，如果矩陣比較大也可以直接用 MATLAB 的 gpuArray 指令，來用 GPU 加速運算，使用起來非常方便，但是到了 Python 就沒有內建這些加速功能，好在 Python 的優點就是有非常完整的套件可以幫助我完成一些原本在 MATLAB 的工作。 這邊會介紹兩種工具，分別是 CuPy 以及 PyVkFFT。 CuPy CuPy 是兼容 NumPy、SciPy 的 GPU 加速函式庫。簡單講就是 NumPy 的數學運算都可以直接轉換成 CuPy 的 API，所以要將原有的程式改寫成 GPU 加速會 …