TensorFlowをコンパイルできません

Question

TensorFlowをソースからコンパイルしようとすると、次のエラーが発生します。 ご意見があれば助かります。

bazelアウト/ホスト/ binに/ _solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so：未定義：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アウト/ホスト/ binに/ _solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.soへの未定義参照 bazelアウト/ホスト/ binに/ _solib_local：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' 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アウト/ホスト/ binに/ _solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.soへの言及/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc___Utensorflow/libtensorflow_framework.so：undefine bazelアウト/ホスト/ binに/：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' 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アウト/ホスト/ binに/ _solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.soとd参照_solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so：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アウト/ホスト/ binに/ _solib_local/_U_S_Stensorflow_Spython_Cgen_Unn_Uops_Upy_Uwrappers_Ucc ___ Utensorflow/libtensorflow_framework.so：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

私たちのbuiにバグがあるようですld。私は自分のマシンで同じものを再現することができました。 LD_LIBRARY_PATHという値が、bazelビルド時に必ずしも正しく伝播されないように見えます。私の場合は、このコマンドを使用したときに正常にビルドできました：

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