为什么可以使用多个值设置环境变量TF_CUDA_COMPUTE_CAPABILITIES？

Question

环境变量TF_CUDA_COMPUTE_CAPABILITIES可以设置为：TF_CUDA_COMPUTE_CAPABILITIES = 3.0,3.5,5.2,7.0,7.5，为什么可以使用多个值进行设置？

Answer 1

我想是为了在编译时适合不同的CUDA版本。

Answer 2

TF_CUDA_COMPUTE_CAPABILITIES是TensorFlow的构建时间参数，如果您在运行时进行设置，则无效。

通过将其设置为多个值，您可以使用cuda库来构建支持多种GPU的TensorFlow安装映像，以利用图形卡硬件的功能。

您包含的TF_CUDA_COMPUTE_CAPABILITIES值越多，构建所需的时间就越长，最终的车轮尺寸就越大。

使用此网页：https://developer.nvidia.com/cuda-gpus#compute，您可以找到要使用的每个图形卡的cuda计算功能值。例如，如果您使用P100，则计算能力为6.0。

继续该示例，如果您仅使用计算能力6.0构建TensorFlow，然后将其安装在具有V100（计算能力7.0）的系统上，则可以运行，但运行速度不如包含计算能力的构建速度快7.0。