pypy安装,pycuda编程

  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的矩阵乘法

pypy安装,pycuda编程