Build 配置Tensorflow时,我应该在“计算能力列表”字段中输入什么?

Build 配置Tensorflow时,我应该在“计算能力列表”字段中输入什么?,build,configuration,tensorflow,Build,Configuration,Tensorflow,我正在从源代码构建Tensorflow,它要求我 Please specify a list of comma-separated Cuda compute capabilities you want to build with. You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus. Please note that each additional comput

我正在从源代码构建Tensorflow,它要求我

Please specify a list of comma-separated Cuda compute capabilities you want to build with.
You can find the compute capability of your device at: https://developer.nvidia.com/cuda-gpus.
Please note that each additional compute capability significantly increases your build time and binary size.
[Default is: "3.5,5.2"]:

我发现这与我的卡片计算能力有关,5.0,顺便说一句,但显然这不是它想要的,因为它是一个数字,而不是一个列表。我找不到任何解释,也没有足够的技能从源代码中理解它。我应该在这里写些什么?

总结这些评论作为答案:

您可以将5.0放在那里,这是您的卡支持的计算功能,应该是您的最佳选择。3.5太低不能使用您的卡的全部功能,5.2太高不受您的卡支持;无论是哪种情况,我相信二进制代码都会在GPU上执行之前用5.0及时重新编译。如果直接使用5.0,它将不会触发即时编译,而是使用更优化的脱机编译

如果您提供一个列表,例如3.5、5.0、5.2,它将脱机编译所有三种计算功能,并允许脱机编译的优化代码在支持3.5、5.0或5.2的GPU卡上运行,而不会触发优化程度较低的即时编译。将更多计算功能放在列表中的成本是增加的脱机编译时间和二进制大小。

转到


找到你的GPU。你会发现计算能力,就像我的是6.1,你能把5.0放在那里吗?我想我能;但这样的随机猜测比默认猜测好吗?我认为默认值是有原因的,除非我理解我所做的,否则我宁愿不更改它。计算能力基于您的卡,即5.0。这应该是你最好的选择。3.5太低不能使用您的卡的全部功能,5.2太高不受您的卡支持;在任何一种情况下,我相信二进制代码都会在GPU上执行之前及时用5.0重新编译。如果你直接使用5.0,它不会触发即时编译,而是会使用更优化的离线编译。@YaoZhang,哦,好吧,我想我明白了。介意把它作为一个答案,这样我就可以接受了吗?顺便问一下,如果使用确切的值更好的话,为什么它是一个列表呢?