尝试从源代码构建TensorFlow时TF_CUDA_COMPUTE_CAPABILITIES变量的含义是什么

时间:2018-10-16 12:02:38

标签: tensorflow cuda nvidia nvidia-docker

从源代码为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架构为目标意味着什么?

0 个答案:

没有答案