diff options
author | Vijay Vasudevan <vrv@google.com> | 2016-02-18 11:22:01 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-02-18 11:28:30 -0800 |
commit | 07fa55268fc63a04cab81508e5468ee1b1914448 (patch) | |
tree | f59cf413e02f2b7cc345df8cecc2c0b6d8d5dcbe /tensorflow/stream_executor/dso_loader.cc | |
parent | 12a8fadb3af52a2105cb09d43a6fad2cf0fda34d (diff) |
TensorFlow: fix missing GetCudaVersion() for libcuda.so. h/t to @3XX0.
Change: 114983764
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.cc')
-rw-r--r-- | tensorflow/stream_executor/dso_loader.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc index fefea9144d..bd5e2c2450 100644 --- a/tensorflow/stream_executor/dso_loader.cc +++ b/tensorflow/stream_executor/dso_loader.cc @@ -69,7 +69,7 @@ string GetCudnnVersion() { return ""; } } /* static */ port::Status DsoLoader::GetLibcudaDsoHandle(void** dso_handle) { - return GetDsoHandle(FindDsoPath("libcuda.so", + return GetDsoHandle(FindDsoPath("libcuda.so" + GetCudaVersion(), "third_party/gpus/cuda/driver/lib64"), dso_handle); } |