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