An error message during installation is brought by the commit 57460f0, on the following platform: A100 with Driver Version: 520.61.05 and CUDA Version: 11.8. I install PyTorch with version 2.2.0 and cuda 11.8. The error message is the following:
nvcc fatal : Unsupported gpu architecture 'compute_90'
ninja: build stopped: subcommand failed.