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, 3 insertions, 4 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 index 64ebc4607a..9bcc3925a8 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu-cuda9-cudnn7 @@ -101,12 +101,11 @@ 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 - -# Clean up pip wheel and Bazel cache when done. -RUN pip --no-cache-dir install --upgrade /pip_pkg/tensorflow-*.whl && \ + bazel-bin/tensorflow/tools/pip_package/build_pip_package /pip_pkg && \ + 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 |