pypy安装,pycuda编程
pyCUDA 1的安装和使用。安装(1)找到适合你的Python解释器和cuda的pycuda版本。
在这个网站上下载网站找到Pycuda
网站主页
找到皮库达
例如pycuda 2020.1 Cu 101 CP 37 CP 37 mwin _ amd64 . whl。
Pycuda 2020.1意味着pycuda版本是2020.1。
Cuda101表示对应的Cuda版本为10.1。
Cp37意味着Python解释器版本是3.7。
只要找到对应的版本,
这里可以不下载
,用pip命令下载安装即可。(2)安装
pip安装pycuda==2020.1
就是这样!
* *注:* *必须对应自己版本的CUDA和Python。
我是python3.7.6和CUDA10.1。
2.简单用进口pycuda。auto init导入pycuda。驱动as cuda从pycuda.com piler导入sourcemoduleimport numpy as NP导入时间#内核写mod=SourceModule( 自带内核函数)函数名=mod.get_function(自写内核函数名)if _ _ name _= _ main _ _: #设置矩阵规范n=10000 block_size=16 #设置网格和块block=(block_size,block _ size,1) if n% block _ size!=0: grid=(n//block _ size 1,n//block _ size 1,1) else: grid=(n//block _ size,n//block _ size,1) # Execute GPU程序函数名(参数)Mine是pycuda的矩阵乘法
运行成功!
1000 * 10000的矩阵乘法