Skip to content

Commit

Permalink
Add GPG key update patch
Browse files Browse the repository at this point in the history
  • Loading branch information
Tiryoh committed Jun 30, 2022
1 parent 4e43846 commit b4452e0
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion docker/tensorflow/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,12 @@
FROM tensorflow/tensorflow:2.8.0-gpu-jupyter

# Update the CUDA Linux GPG Repository Key
# https://github.com/NVIDIA/nvidia-docker/issues/1631#issuecomment-1112828208
RUN rm /etc/apt/sources.list.d/cuda.list && \
rm /etc/apt/sources.list.d/nvidia-ml.list && \
apt-key del 7fa2af80 && \
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN apt-get update && \
DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y sudo && \
apt-get clean && rm -rf /var/lib/apt/lists/*
Expand All @@ -9,7 +16,7 @@ RUN useradd --create-home --shell /bin/bash --user-group --groups adm,sudo,audio
RUN mkdir /.jupyter && chmod 777 /.jupyter && \
mkdir /.cache && chmod 777 /.cache && \
mkdir /.config && chmod 777 /.config
RUN python3 -m pip install jupyterlab pandas pillow
RUN python3 -m pip install jupyterlab pandas pillow scipy sklearn seaborn
RUN python3 -m pip install keras_cv_attention_models

CMD [ "bash", "-c", "source /etc/bash.bashrc && /usr/local/bin/jupyter lab --notebook-dir=/tf --ip 0.0.0.0 --no-browser --allow-root" ]

0 comments on commit b4452e0

Please sign in to comment.