Skip to content

Commit f74a764

Browse files
committed
Do not install tensorflow 2.18.0 as it gives an error
1 parent a1265be commit f74a764

File tree

2 files changed

+1
-8
lines changed

2 files changed

+1
-8
lines changed

images/tensorflow-notebook/cuda/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
1414
# Install TensorFlow, CUDA and cuDNN with pip
1515
RUN pip install --no-cache-dir \
1616
"jupyter-server-proxy" \
17-
"tensorflow[and-cuda]" && \
17+
"tensorflow[and-cuda]<=2.17.1" && \
1818
fix-permissions "${CONDA_DIR}" && \
1919
fix-permissions "/home/${NB_USER}"
2020

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,6 @@
11
# Copyright (c) Jupyter Development Team.
22
# Distributed under the terms of the Modified BSD License.
3-
import os
4-
53
import tensorflow as tf
64

7-
# Suppress logging warnings
8-
# https://stackoverflow.com/a/78803598
9-
os.environ["GRPC_VERBOSITY"] = "ERROR"
10-
os.environ["GLOG_minloglevel"] = "2"
11-
125
print(tf.constant("Hello, TensorFlow"))
136
print(tf.reduce_sum(tf.random.normal([1000, 1000])))

0 commit comments

Comments
 (0)