提交 0685152a 编写于 作者: A Alex Rothberg 提交者: Vijay Vasudevan

Hard-code libcuda version number to "1". Fixes #2865. (#3576)

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").
上级 cbfd2787
......@@ -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);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册