search

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 450 forks source link

Unsupported gpu architecture #407

Open holyseven opened 2 years ago

holyseven commented 2 years ago

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 commented 2 years ago

Are you solve it?

zhanghang1989 commented 2 years ago

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