diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 index 9bcc3925a8..64ebc4607a 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 @@ -101,11 +101,12 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib --jobs=${TF_AVAILABLE_CPUS} \ tensorflow/tools/pip_package:build_pip_package && \ mkdir /pip_pkg && \ - bazel-bin/tensorflow/tools/pip_package/build_pip_package /pip_pkg && \ - pip --no-cache-dir install --upgrade /pip_pkg/tensorflow-*.whl && \ + bazel-bin/tensorflow/tools/pip_package/build_pip_package /pip_pkg + +# Clean up pip wheel and Bazel cache when done. +RUN pip --no-cache-dir install --upgrade /pip_pkg/tensorflow-*.whl && \ rm -rf /pip_pkg && \ rm -rf /root/.cache -# Clean up pip wheel and Bazel cache when done. WORKDIR /root |