Python GPU5:Pycuda例子

例子


import pycuda.gpuarray as gpuarray
import pycuda.driver as cuda
import pycuda.autoinit
import numpy

a_gpu = gpuarray.to_gpu(numpy.random.randn(4,4).astype(numpy.float32))
a_doubled = (2*a_gpu).get()
print (a_doubled)
print (a_gpu6

英文文档

Tutorial - pycuda 2021.1 documentation

Before you can use PyCuda, you have to import and initialize it: Note that you do not have to use - initialization, context creation, and cleanup can also be performed manually, if desired. The next step in most programs is to transfer data onto the device.

https://documen.tician.de/pycuda/tutorial.html

中文文档

教程 - PyCUDA 2020.1 文档

https://www.osgeo.cn/pycuda/tutorial.html