TensorFlow kann nicht kompiliert werden

Question

Die folgenden Fehler werden beim Versuch, TensorFlow aus der Quelle zu kompilieren, angezeigt. Irgendwelche Gedanken wären hilfreich.

bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezug auf cublasGemmEx@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to cublasZhpmv_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezugnahme auf cufftExecD2Z@libcufft.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to cublasSrotg_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezug auf cufftExecR2C@libcufft.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to cublasSsyrk_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local /_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefiniert d Bezugnahme auf cublasDgemm_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to cufftSetWorkArea@libcufft.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezug auf cublasChemm_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to cublasZher2k_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezug auf cufftExecC2C@libcufft.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to curandSetStream@libcurand.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so: undefined Bezug auf cublasDrotm_v2@libcublas.so.9.0' bazel-out/host/bin/_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so: undefined reference to curandSetPseudoRandomGeneratorSeed@libcurand.so.9.0'

Answer 1

Es scheint einen Fehler in unserem bui zu geben ld. Ich konnte dasselbe auf meiner Maschine reproduzieren. Sieht so aus, als ob der Wert LD_LIBRARY_PATH während des Bazel-Builds nicht immer korrekt weitergegeben wird. In meinem Fall konnte ich erfolgreich bauen, wenn ich diesen Befehl verwendete:

bazel build --config=opt --config=cuda tensorflow/tools/pip_package:build_pip_package --action_env="LD_LIBRARY_PATH=${LD_LIBRARY_PATH}"