Unsupported gpu architecture

[holyseven]

Environment: Ubuntu 18.04, CUDA 11.0, PyTorch 1.7.1, A100.

When installed from source by python setup.py install, an nvcc fatal error raised:

nvcc fatal   : Unsupported gpu architecture 'compute_80'

RuntimeError: Error compiling objects for extension

[ZhouZoey]

Are you solve it?

[zhanghang1989]

I don't have access to a recent gpu, but this sounds like a pytorch issue. Could you try asking questions at https://github.com/pytorch/extension-cpp