Unable to import tinycudann during docker run. (OSError: Unknown compute capability. Ensure PyTorch with CUDA support is installed)

georghess / neurad-studio

[CVPR2024] NeuRAD: Neural Rendering for Autonomous Driving

https://research.zenseact.com/publications/neurad/

Apache License 2.0

346 stars 24 forks source link

Unable to import tinycudann during docker run. (OSError: Unknown compute capability. Ensure PyTorch with CUDA support is installed) #40

Closed Anji-Builds closed 3 months ago

Anji-Builds commented 4 months ago

Thanks a lot for this interesting work.

I used the provided Dockerfile with slight modification and was able to build the docker image. However, when I run the docker container and try to import tinycudann, I get the above error. I have provided the TCNN_CUDA_ARCHITECTURES value.

Could anyone help me solve this issue?

Andyshen555 commented 4 months ago

I'm not a docker expert but the warning says your Nvidia driver is not detected. Either you don't have nvidia driver in your docker, or most likely your docker doesn't have access to your nvidia GPU.

atonderski commented 3 months ago

thanks Andyshen, closing due to lack of further response

Anji-Builds commented 3 months ago

Thank you @Andyshen555 for your insight. You were right.

@atonderski yes the issue is solved.