Cuda Python supported on Jetson TX2?

Autonomous Machines Jetson & Embedded Systems Jetson TX2
1

Hi,

Is Cuda python supported on Jetson TX2 or Jetson series?

I read, Numba: High-Performance Python with CUDA Acceleration.
https://devblogs.nvidia.com/numba-python-cuda-acceleration/

2

Hi,

Please install pycuda to enable CUDA python support.

$ sudo apt-get install python3-pip 
$ pip3 install pycuda --user

Not all the python module are supported on the Jetson.
For Numba, you will need to build LLVM from source first.
Please find this topic for more information:
https://devtalk.nvidia.com/default/topic/1028001/jetson-tx1/numba-package-in-python-on-jetson-/post/5323156/#5323156

Thanks.

3

Hi AastaLLL

It works, thanks!