diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc index fab3385cca2966e52614b65cf863e83c0a2045a6..cce31ef4dcf7123b610c3f724bf8753b5d1625f1 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); }