遇到这种问题:很可能是cuda编译运算的架构,与你当前的显卡不符合。虽然可以编译通过,但是运行时却会出现错误。
解决办法:
'--gpu-architecture=compute_61', # change compute_70 -> compute_61
'--gpu-code=sm_61', #change sm_70 -> sm_61
遇到这种问题:很可能是cuda编译运算的架构,与你当前的显卡不符合。虽然可以编译通过,但是运行时却会出现错误。
解决办法:
'--gpu-architecture=compute_61', # change compute_70 -> compute_61
'--gpu-code=sm_61', #change sm_70 -> sm_61