从源代码为GPU构建TensorFlow 1.11.0时,我们可以修改环境变量GroupBy
来定位特定版本的GPU体系结构,如下所述:https://docs.nvidia.com/deeplearning/dgx/tensorflow-user-guide/index.html#tfaddparam
TensorFlow https://storage.googleapis.com/tensorflow-kokoro-build-badges/ubuntu-gpu-py3.html的官方Google构建 将此变量设置为6.0,表示它们针对的是Pascal架构。
到目前为止,我可以使用Nvidia V100 Volta在工作站上安装GPU的官方TF,并且可以正常工作。
所以我的问题是,在这种情况下以特定的GPU架构为目标意味着什么?