mirror of
https://github.com/PacktPublishing/Hands-On-GPU-Programming-with-CUDA-C-and-Python-3.x-Second-Edition.git
synced 2025-07-21 04:41:05 +02:00
13 lines
200 B
Python
13 lines
200 B
Python
import numpy as np
|
|
from pycuda import gpuarray
|
|
import pycuda.driver as drv
|
|
|
|
drv.init()
|
|
dev = drv.Device(0)
|
|
ctx = dev.make_context()
|
|
|
|
x = gpuarray.to_gpu(np.float32([1,2,3]))
|
|
print(x.get())
|
|
|
|
ctx.pop()
|