OSError: Could not find compatible tinycudann extension for compute capability 87. about tiny-cuda-nn HOT 5 CLOSED

Hopefully this workaround is no longer needed. I just pushed an attempted fix that should make it work out of the box (see also #436). Please try re-downloading and installing again -- could you let me know if it works?

from tiny-cuda-nn.

I managed to resolve it by rebuiling tinycudann, the key step is to export TCNN_CUDA_ARCHITECTURES=86 before intalling PyTorch extension.

I think tinycudann dose not support Jetson AGX Orin yet, because the jetson's cuda capability (87) is not in the supported list

import tinycudann
/home/xns/miniconda3/envs/3dgs/lib/python3.10/site-packages/tinycudann-1.7-py3.10-linux-aarch64.egg/tinycudann/modules.py:53: UserWarning: tinycudann was built for lower compute capability (86) than the system's (87). Performance may be suboptimal.
  warnings.warn(f"tinycudann was built for lower compute capability ({cc}) than the system's ({system_compute_capability}). Performance may be suboptimal.")

from tiny-cuda-nn.

Hello,
I have the same problem on my jetson orin, do you install tinycudann before instaling pytorch then? and can you elaborate in which step exactly you do : export TCNN_CUDA_ARCHITECTURES=86 ?
Thank you

from tiny-cuda-nn.

Hopefully this workaround is no longer needed. I just pushed an attempted fix that should make it work out of the box (see also #436). Please try re-downloading and installing again -- could you let me know if it works?

Thanks for your reply, I will check it later！

from tiny-cuda-nn.

It works well now。

Hopefully this workaround is no longer needed. I just pushed an attempted fix that should make it work out of the box (see also #436). Please try re-downloading and installing again -- could you let me know if it works?

Thanks for your reply, I will check it later！

from tiny-cuda-nn.