Replies: 1 comment
-
Does this example help? https://github.com/inducer/pycuda/blob/main/examples/cai_cupy_arrays.py |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi, I try to use pycuda to create pinned memory for spped up the transfer speed from cpu to gpu. I can btain a gpuarray in gpu after doing the
cuda.memcpy_htod(devPtr, img_pin)
gpu_array = gpuarray.GPUArray(test_img_trasform.shape, np.float32, gpudata=devPtr)
How can I transfer this gpu_array to tensor in gpu?
Thanks in adcanced.
Beta Was this translation helpful? Give feedback.
All reactions