From f45874b4a969a60fa9ced86e6769012e4912b5ca Mon Sep 17 00:00:00 2001 From: Bob Adolf Date: Tue, 14 Jun 2016 21:54:56 -0400 Subject: [PATCH] Hard-code libcuda version number to "1". Fixes #2865. As per comments from nvidia-docker dev @3XX0, hardcoding "1" should be reasonably safe. The TF_CUDA_VERSION variable from the configure script is not appropriate here (it will contain something like "7.0" or "7.5", while the libcuda soname major version number should be "1"). --- tensorflow/stream_executor/dso_loader.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc index fab3385cca2966..cce31ef4dcf712 100644 --- a/tensorflow/stream_executor/dso_loader.cc +++ b/tensorflow/stream_executor/dso_loader.cc @@ -72,7 +72,7 @@ string GetCudnnVersion() { return ""; } } /* static */ port::Status DsoLoader::GetLibcudaDsoHandle(void** dso_handle) { - return GetDsoHandle(FindDsoPath(tensorflow::internal::FormatLibraryFileName("cuda", ""), + return GetDsoHandle(FindDsoPath(tensorflow::internal::FormatLibraryFileName("cuda", "1"), GetCudaDriverLibraryPath()), dso_handle); }