Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Fix] Docker not supporting older CUDA architectures #1328

Conversation

Zunhammer
Copy link
Contributor

@Zunhammer Zunhammer commented Feb 1, 2023

Added the most common CUDA architectures to the Dockerfile as well as the docker image found at https://hub.docker.com/r/dromni/nerfstudio/tags
Also improved the description in the docs.
Now the following CUDA architectures should work:

GPU CUDA arch
H100 90
40X0 89
30X0 86
A100 80
20X0 75
TITAN V / V100 70
10X0 / TITAN Xp 61
9X0 52
K80 37

Fixes #1317
Fixes #1056
Fixes #1168

Copy link
Contributor

@tancik tancik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks for adding the extra info

@Zunhammer Zunhammer merged commit 9966c43 into nerfstudio-project:main Feb 1, 2023
chris838 pushed a commit to chris838/nerfstudio that referenced this pull request Apr 22, 2023
…ect#1328)

Fix docker not supporting older CUDA architectures.

Co-authored-by: Nicolas Zunhammer <[email protected]>
Co-authored-by: Matthew Tancik <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants