Skip to content

Commit fcd0619

Browse files
committed
downgrade cudnn from 8 to 7
1 parent 34ad1c5 commit fcd0619

File tree

2 files changed

+25
-11
lines changed

2 files changed

+25
-11
lines changed

commons/Dockerfile.base.22.09-py36-cuda10.2

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
FROM nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04
1+
FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04
22

33
# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
44
ARG CUDA=10.2
55
ARG LIBNVINFER=7.1.3-1
66
ARG LIBNVINFER_MAJOR_VERSION=7
77
ARG CUDNN=7.6.5.32-1
88
ENV NCCL_VERSION=2.7.8-1+cuda10.2
9-
ENV CUDNN_VERSION 8.0.2.39
9+
ENV CUDNN_VERSION 7.6.5.32
1010

1111
# Python 2.7 or 3.6 is supported by Ubuntu Bionic out of the box
1212
ARG python=3.6
@@ -23,7 +23,10 @@ ENV DEBIAN_FRONTEND=noninteractive \
2323
# Set default shell to /bin/bash
2424
SHELL ["/bin/bash", "-cu"]
2525

26-
RUN apt-get update -y && \
26+
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \
27+
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
28+
RUN \
29+
apt-get update -y && \
2730
apt-get install -y --no-install-recommends software-properties-common && \
2831
add-apt-repository -y "deb http://security.ubuntu.com/ubuntu xenial-security main" && \
2932
apt-get update -y && \

python-tensorflow/Dockerfile.2.8-py38-cuda11.3

Lines changed: 19 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,16 +11,27 @@ RUN python3 -m pip install --no-cache-dir \
1111
proto-plus==1.10.2
1212
RUN python3 -m pip install --no-cache-dir \
1313
--use-feature=2020-resolver \
14-
tensorflow-transform==0.25.0 \
15-
tensorflow_model_analysis \
16-
tensorflow-datasets \
17-
tensorflow_probability==0.11.1 \
14+
tensorflow-transform==1.10.1 \
15+
tensorflow_model_analysis==0.41.1 \
16+
tensorflow-datasets==4.7.0 \
17+
tensorflow_probability==0.17.0 \
1818
tensorflow-addons==0.11.2 \
1919
neural-structured-learning==1.3.1 \
2020
tensorflow-model-optimization==0.5.0 \
21-
tfx==0.25.0 \
22-
tfx-bsl==0.25.0 \
23-
tf-agents==0.6.0
21+
# tensorflow-transform==0.25.0 \
22+
# tensorflow_model_analysis \
23+
# tensorflow-datasets \
24+
# tensorflow_probability==0.11.1 \
25+
# tensorflow-addons==0.11.2 \
26+
neural-structured-learning==1.3.1 \
27+
tensorflow-model-optimization==0.5.0 \
28+
# tensorflow-model-optimization==0.5.0 \
29+
tfx==1.10.0 \
30+
tfx-bsl==1.10.1 \
31+
tf-agents==0.14.0
32+
# tfx==0.25.0 \
33+
# tfx-bsl==0.25.0 \
34+
# tf-agents==0.6.0
2435

2536
RUN python3 -m pip install --no-cache-dir \
2637
--use-feature=2020-resolver \
@@ -51,7 +62,7 @@ RUN python3 -m pip install --no-cache-dir \
5162
tensorflow-plot==0.3.2 \
5263
tensorflow_text==2.8.2 \
5364
tensorflow-gan==2.0.0
54-
RUN python3 -m pip install --no-cache-dir pyfarmhash==0.3.2 pyarrow==8.0.0 && \
65+
RUN python3 -m pip install --no-cache-dir pyfarmhash==0.3.2 pyarrow==5.0.0 && \
5566
python3 -m pip install --no-cache-dir \
5667
-i https://pypi-nightly.tensorflow.org/simple tensorflow-data-validation==1.8.0.dev20220515
5768

0 commit comments

Comments
 (0)