hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-13 21:12:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not install tensorflow 2.18.0 as it gives an error

Browse Source
This commit is contained in:
Ayaz Salikhov
2024-10-28 14:06:50 +00:00
parent a1265be650
commit f74a764584
2 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
images/tensorflow-notebook/cuda/Dockerfile
View File

@@ -14,7 +14,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
# Install TensorFlow, CUDA and cuDNN with pip # Install TensorFlow, CUDA and cuDNN with pip
RUN pip install --no-cache-dir \ RUN pip install --no-cache-dir \
"jupyter-server-proxy" \ "jupyter-server-proxy" \
"tensorflow[and-cuda]" && \ "tensorflow[and-cuda]<=2.17.1" && \
fix-permissions "${CONDA_DIR}" && \ fix-permissions "${CONDA_DIR}" && \
fix-permissions "/home/${NB_USER}" fix-permissions "/home/${NB_USER}"

7
tests/tensorflow-notebook/units/unit_tensorflow.py
View File

@@ -1,13 +1,6 @@
# Copyright (c) Jupyter Development Team. # Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License. # Distributed under the terms of the Modified BSD License.
import os
import tensorflow as tf import tensorflow as tf
# Suppress logging warnings
# https://stackoverflow.com/a/78803598
os.environ["GRPC_VERBOSITY"] = "ERROR"
os.environ["GLOG_minloglevel"] = "2"
print(tf.constant("Hello, TensorFlow")) print(tf.constant("Hello, TensorFlow"))
print(tf.reduce_sum(tf.random.normal([1000, 1000]))) print(tf.reduce_sum(tf.random.normal([1000, 1000])))