要查看本机GPU的算力,可以通过以下命令在终端中运行:

nvidia-smi

这将显示GPU的详细信息,包括计算能力。例如,计算能力为6.1表示该GPU支持CUDA 8.0。

如果要更改代码中的-gencode选项以匹配您的GPU的计算能力,请在编译代码时使用以下命令:

nvcc -gencode=arch=compute_61,code=sm_61 example.cu

在这个例子中,我们将-gencode选项设置为符合计算能力为6.1的GPU。请注意,我们还将代码选项设置为sm_61,这是根据计算能力确定的。

然后,您可以使用以下命令运行代码:

./example

这将使用匹配您GPU的计算能力的代码进行计算。

ubantu如何查看本机GPU算力并更改代码 -gencode arch=compute_61code=compute_61

原文地址: https://www.cveoy.top/t/topic/eHBJ 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录