diff options
| author |  Allen Lavoie <allenlavoie@gmail.com> | 2017-10-02 00:19:23 -0700 |
|---|---|---|
| committer |  gunan <gunan@google.com> | 2017-10-02 00:19:23 -0700 |
| commit | e3ceea3f65a4091b2a13f3e9c34bf4d1cf3c27fe (patch) | |
| tree | 325ae59fed5c3f4ac4f69c4f151009c1c01251e9 /tensorflow/tools/docker/Dockerfile.devel-gpu | |
| parent | 242b0f11302095110ba4014b62a4ca98833d827e (diff) | |
Fix the Docker GPU build (adds a symlink + library path) (#13399)
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index f5364d803a..04773376e9 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -78,10 +78,12 @@ WORKDIR /tensorflow # Configure the build for our CUDA configuration. ENV CI_BUILD_PYTHON python -ENV LD_LIBRARY_PATH /usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH +ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH ENV TF_NEED_CUDA 1 ENV TF_CUDA_COMPUTE_CAPABILITIES=3.0,3.5,5.2,6.0,6.1 +RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 + RUN tensorflow/tools/ci_build/builds/configured GPU \ bazel build -c opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" \ tensorflow/tools/pip_package:build_pip_package && \ |