Skip to content

Commit

Permalink
Merge pull request #1485 from mathbunnyru/asalikhov/fix_tensorflow
Browse files Browse the repository at this point in the history
Fix tensorflow keras bug
  • Loading branch information
mathbunnyru authored Oct 16, 2021
2 parents 7c05535 + 8f06d5c commit ef5e9dd
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 0 deletions.
1 change: 1 addition & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ repos:
rev: v2.7.0
hooks:
- id: hadolint-docker
entry: hadolint/hadolint:v2.7.0 hadolint

# Lint: YAML
- repo: https://github.com/adrienverge/yamllint.git
Expand Down
1 change: 1 addition & 0 deletions r-notebook/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ RUN mamba install --quiet --yes \
fix-permissions "/home/${NB_USER}"

# These packages are not easy to install under arm
# hadolint ignore=SC2039
RUN set -x && \
arch=$(uname -m) && \
if [ "${arch}" == "x86_64" ]; then \
Expand Down
10 changes: 10 additions & 0 deletions tensorflow-notebook/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,19 @@ FROM $BASE_CONTAINER

LABEL maintainer="Jupyter Project <jupyter@googlegroups.com>"

# Fix DL4006
SHELL ["/bin/bash", "-o", "pipefail", "-c"]

# Install Tensorflow
RUN mamba install --quiet --yes \
'tensorflow' && \
mamba clean --all -f -y && \
fix-permissions "${CONDA_DIR}" && \
fix-permissions "/home/${NB_USER}"

# Temporary fix for https://github.com/jupyter/docker-stacks/issues/1480
USER root
WORKDIR /opt/conda/lib/python3.9/site-packages/
RUN wget -qO- https://patch-diff.githubusercontent.com/raw/tensorflow/tensorflow/pull/51450.diff | git apply
USER ${NB_UID}
WORKDIR "${HOME}"

0 comments on commit ef5e9dd

Please sign in to comment.