pycuda使用方法

Step1 安装visual studio，然后安装新版本的cuda，比如cuda11

Step2  安装pycuda, 命令如下：pip install pycuda

Step3 运行以下例子：

import pycuda
import pycuda.autoinit
import pycuda.driver as drv
import numpy
from pycuda.compiler import SourceModule
### 防止找不到c语言编译器 
import os
if (os.system("cl.exe")):
    os.environ['PATH'] += ';'+r"D:\Program Files\Microsoft Visual Studio\2022\Community\VC\Tools\MSVC\14.16.27023\bin\HostX64\x64"
if (os.system("cl.exe")):
    raise RuntimeError("cl.exe still not found, path probably incorrect")
##### 具体乘法模块 
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
  const int i = threadIdx.x;
  dest[i] = a[i] * b[i];
}
""")
multiply_them = mod.get_function("multiply_them")
a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)
dest = n