rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.19k stars 527 forks source link

[QST] How to set cuml for clustering under Jetson Orin NX board based on arm64. #6087

Open anrkq1024 opened 2 weeks ago

anrkq1024 commented 2 weeks ago

First, I appreciate your review.

I tried to set an environments for cuml under jetson orin nx board, using conda. I found that cuda 11.4 version is compatible with cuml 22.12.00. Image Image

So, I could import cuml, pyzed for zed stereo camera.

However, as I tried to apply an array into DBSCAN library of cuml, it caused an error. I searched some blogs to solve it, someone explained that issue is caused by version miss matching between cuda and cuml. But, I think that the libraries in conda env are correctly installed.

Since It is my first try to use cuml under jetson nx board, I spent many days for this issue. I hope that you suggest any method, which I can try in order to solve it.

Code Image

Error Image

divyegala commented 2 weeks ago

@anrkq1024 we do not publish conda packages that are compiled for jetson orin nx board. I would recommend trying to compile from source.

quasiben commented 2 weeks ago

@anrkq1024, If possible can you share a bit about how you might be using cuML on Jetson boards ? Are you training (or training and infering) in an IoT like environment ?