Python提供了一個 名爲PyCUDA 的封裝，通過使用Nvidia CUDA API能夠利用GPU的並行計算能力; Python 也有一個PyOpenCL模塊能夠用於任何GPU的並行計算。
而PyCUDA是主要針對NVIDIA的GPU的，爲了使用該庫，需要你有相應的GPU設備，以及安裝了CUDA toolkit （CUDA toolkit下載地址）；
Step1 安裝PyCUDA
安裝前提是確保你裝了python，cuda；
window下 到PyCUDA下載鏈接 下載對應CUDA版本，python版本，系統版本的安裝文件（以.whl結尾的文件）；
推薦使用anaconda環境，然後打卡個prompt終端，用pip 安裝上面下載的whl文件；
pip istall xxxxx-your-download-file-xxx.whl
爲了確保安裝完成，用ipython打開，輸入
import pycuda
看是否報錯，如沒有報錯則表示安裝完成；
Linux 下安裝的方式
我這邊依然推薦anaconda環境管理；
打開個終端，運行如下指令：
conda install -c l