admin管理员组

文章数量:1663034

打开 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')

本文标签: nvccFATALMegatronLMArchitecture