diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2016-02-18 12:25:23 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-02-18 12:27:19 -0800 |
| commit | d0a822fbcb04d95a643d8efe65699a8d1cdce98b (patch) | |
| tree | 1e21b4b744f492ea2e97fe8ac73c881900c26341 /tensorflow/stream_executor/dso_loader.cc | |
| parent | 16f46d932c4e25773003c580924f8c718f2ad9c1 (diff) | |
TensorFlow: switch dsoloader path for libcuda to end in .so.1
Change: 114990321
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.cc')
| -rw-r--r-- | tensorflow/stream_executor/dso_loader.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc index bd5e2c2450..caf10a9003 100644 --- a/tensorflow/stream_executor/dso_loader.cc +++ b/tensorflow/stream_executor/dso_loader.cc @@ -69,9 +69,9 @@ string GetCudnnVersion() { return ""; } } /* static */ port::Status DsoLoader::GetLibcudaDsoHandle(void** dso_handle) { - return GetDsoHandle(FindDsoPath("libcuda.so" + GetCudaVersion(), - "third_party/gpus/cuda/driver/lib64"), - dso_handle); + return GetDsoHandle( + FindDsoPath("libcuda.so.1", "third_party/gpus/cuda/driver/lib64"), + dso_handle); } /* static */ port::Status DsoLoader::GetLibcuptiDsoHandle(void** dso_handle) { |