Skip to content

Commit

Permalink
Revert change of testing logic in gpu.Dockerfile (#167)
Browse files Browse the repository at this point in the history
The nightly build machines don't have a GPU, so this test fails and
Docker images are not getting uploaded.

Reverting the change should fit it for now.
  • Loading branch information
beckerhe authored Nov 20, 2023
1 parent 9f81388 commit fdd023e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tensorflow_runtime_dockerfiles/gpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -52,5 +52,5 @@ CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/
FROM base as test

ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH
COPY test.import_gpu.sh /test.import_gpu.sh
RUN /test.import_gpu.sh
COPY test.import_cpu.sh /test.import_cpu.sh
RUN /test.import_cpu.sh

0 comments on commit fdd023e

Please sign in to comment.