Megatron-LM出现nvcc fatal: Unsupported gpu architecture ‘compute_90‘

打开 Megatron-LM/megatron/fused_kernels/__init__.py，注释掉以下三行即可：

#        if int(bare_metal_minor) >= 7:
#            cc_flag.append('-gencode')
#            cc_flag.append('arch=compute_90,code=sm_90')