使用pip安装TensorFlow很容易,但是用户不知道TensorFlow是如何构建的。有什么办法可以知道预建TensorFlow的所有配置选项吗?我问这个问题是因为我不想从源代码安装TensorFlow,但是我正在使用具有Turing架构的最新GPU,并且具有7.5的计算能力。我使用“ pip install tf-nightly-gpu”安装了TensorFlow,但我不知道预编译的版本是否已编译为支持计算能力为7.5的GPU。
从源代码构建时,我们可以指定所需的计算能力:
请指定要使用的逗号分隔的Cuda计算功能列表。您可以在https://developer.nvidia.com/cuda-gpus上找到设备的计算能力。请注意,每一项额外的计算功能都会显着增加构建时间和二进制文件大小。 [默认值为:3.5,7.0] 6.1
但是对于预构建版本,有什么办法让我们知道所有配置选项吗?