isl-org / isl-org/Open3D

Cannot import open3d.cuda on ARM platform - Jetson ORIN

Open
#6,243 7 comments 0 reactions 0 assignees View on GitHub
build/install
Dominant language
C++
Stars
14k
Forks
2.6k
Avg merge
5d 18h
Merged PRs (30d)
6

Description

### Checklist

- [X] I have searched for [similar issues](https://github.com/isl-org/Open3D/issues).
- [X] For Python issues, I have tested with the [latest development wheel](http://www.open3d.org/docs/latest/getting_started.html#development-version-pip).
- [X] I have checked the [release documentation](http://www.open3d.org/docs/release/) and the [latest documentation](http://www.open3d.org/docs/latest/) (for `master` branch).

### Steps to reproduce the issue

#### I first cloned Open3D by:
```
git clone https://github.com/isl-org/Open3D.git
cd Open3D
```

#### Then, I build Open3D (on Ubuntu 20.04, with CUDA 11.5) with:
```
conda activate your-virtual-env

cd Open3D && mkdir build && cd build
cmake -DBUILD_CUDA_MODULE=ON -DBUILD_GUI=ON ..

# Build
make -j$(nproc)
make install-pip-package -j$(nproc)

```

### Error message
No import error or warning was shown. But after looking into package I found that the open3d is reaching the warning below while initializing

```shell

Open3D was built with CUDA support, but an error ocurred while loading the Open3D CUDA Python bindings. This is usually because the CUDA libraries could not be found. Check your CUDA installation. Falling back to the CPU pybind library. Reported error: {os_error}.',
```

### Open3D, Python and System information

```markdown
- Operating system: Ubuntu 20.04 ARM 64
- Python version: Python 3.8.16
- Open3D version: output from python: 0.17.0+6370a58b9
- System architecture: jetson ORIN AGX 32GB
- Is this a remote workstation?: no
- How did you install Open3D?: build from source
- Compiler version (if built from source): gcc 9.4
```

### Additional information

Clearly Open3d is not albe to identify the GPU on jetson. CUDA 11.8 is installed on jetson and pytorch is able to detect GPU on the system but I don't understand why open3d is not able to identify the gpu.

Am I missing any additional drivers to make this work or is this a bug?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.