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

Error while building Dockerfile: "Unsupported gpu architecture" #22

Open
RobertoRoos opened this issue Mar 22, 2022 · 1 comment
Open

Comments

@RobertoRoos
Copy link

While trying to build the Dockerfile (or actually an extended Dockerfile based on this) I get the following error during the building of OpenPose:

nvcc fatal   : Unsupported gpu architecture 'compute_80'

Running Docker on Ubuntu 20.04, with a Titan Xp GPU.

Oddly enough I can compile OpenPose while running the container, but not when done during the building of the Dockerfile.

Not really your problem, but I wanted to document somewhere.

@RobertoRoos
Copy link
Author

RobertoRoos commented Mar 22, 2022

I found I can fix this by setting the CUDA architecture explicitly while compiling during the Docker build:

cmake .. -DBUILD_PYTHON=ON -DCUDA_ARCH="Pascal"

...as My Titan Xp has the Pascal architecture.

Actually, because a bug in OpenPose this command doesn't work, so instead I run:

cmake .. -DBUILD_PYTHON=ON -DCUDA_ARCH=Manual -DCUDA_ARCH_BIN="60 61 62" -DCUDA_ARCH_PTX="61"

Maybe we can make a note of this in the readme?
And I'm confused why other builds don't fail too.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant