如何查看本机GPU算力在适合的位置将原代码 -gencode arch=compute_61code=compute_61中的61对应行替换为本机的GPU算力
要查看本机GPU的算力,可以使用以下命令:
nvidia-smi
这将显示有关本机GPU的详细信息,包括其计算能力。在这个输出中,您可以找到名为“Compute Capability”的列,其中包含GPU的算力。例如,如果您的GPU的计算能力为6.1,它将显示为“6.1”。
要将原代码中的-gencode选项替换为本机GPU的算力,您只需要将61替换为您的GPU的计算能力。例如,如果您的GPU的计算能力为6.1,您可以将-gencode选项替换为:
-gencode arch=compute_61,code=compute_61
请注意,这只适用于使用CUDA编程的情况,如果您使用的是其他编程框架或工具,则可能需要进行其他设置。
原文地址: https://www.cveoy.top/t/topic/eKox 著作权归作者所有。请勿转载和采集!