OpenCL alternative?

We have an image reconstruction package built on OpenCL and are disappointed to see that JAX does not support OpenCL.

What alternatives are there in CUDA? Our preference is for a Python API.

1 Like

Hi,

You can try PyCuda on Jetson.
Thanks.

Thanks for the pointer. Is there a guide to migration from OpenCL to CUDA?

Hi,

You can find the PyCUDA document below:

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

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.