使用Cuda 9.1,CUDNN 7.1在Debian 10上从源代码构建

时间:2018-08-03 09:52:32

标签: tensorflow

尝试使用cuda 9.1.85在debian 10上构建tensorflow 1.9 和cudnn 7.1.4.18

使用gcc-6作为编译器和构建命令时

bazel build --verbose_failures --config=opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"  //tensorflow/tools/pip_package:build_pip_package

我明白了

INFO: From Compiling external/nccl_archive

/src/libwrap.cu.cc:                                                                                    
/usr/lib/cuda/include/cuda_fp16.h(2958): error: identifier "__float2half_rn" is undefined                                                       

/usr/lib/cuda/include/cuda_fp16.h(3000): error: identifier "__float2half_rn" is undefined                                                       

2 errors detected in the compilation of "/tmp/tmpxft_000070b1_00000000-6_libwrap.cu.cpp1.ii".

1 个答案:

答案 0 :(得分:0)

之所以发生这种情况,是因为内部版本发现了以前的cuda 8.0安装遗留下的一些cuda标头,并且在那里未实现这些功能。清理这些内容后,构建成功完成。