显卡cuda8编译错误"__shfl_up“的Tensorflow

Question

tensorflow 1.2.1 CUDA 8.0.61.2

当使用bazel编译tensorflow时，我得到一个错误：

./tensorflow/core/util/cuda_kernel_helper.h(620)：错误:未定义标识符"__shfl“

./tensorflow/core/util/cuda_kernel_helper.h(640)：错误:未定义标识符"__shfl_up“

./tensorflow/core/util/cuda_kernel_helper.h(660)：错误:未定义标识符"__shfl_down“

./tensorflow/core/util/cuda_kernel_helper.h(680)：错误:未定义标识符"__shfl_xor“

4在"/tmp/tmpxft_000010d3_00000000-7_depthtospace_op_gpu.cu.cpp1.ii".的编译过程中检测到错误

如何解决这个问题？

gcc版本6.2.1 20160916 (Red Hat 6.2.1-3) (GCC)使用compute 2.0

Answer 1

这是由于选择了不受支持的计算能力(数字太低)。支持的最低版本为3.0。我在Github上写了个bug。在配置脚本中，它现在应该是fixed，因此不可能选择低于3.0 Cuda计算能力。

Answer 2

尝试从cuda 10.1迁移到cuda 9.1 sudo apt install nvidia-cuda-toolkit，并将gcc-7迁移到gcc-5。到目前为止，在尝试了所有可用的解决方案后，它对我起作用了。