From c7a5e5c10bb48681c77d22bc012bf761cdd928b4 Mon Sep 17 00:00:00 2001 From: Rafael Balthazar Date: Sun, 7 Jan 2024 12:13:02 -0300 Subject: [PATCH] Fix TensorFlow CPU-only package --- docker/Dockerfile.gpu | 2 +- requirements.gpu.txt | 17 +++++++++++++++++ requirements.txt | 2 +- 3 files changed, 19 insertions(+), 2 deletions(-) create mode 100644 requirements.gpu.txt diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu index 93d421f..121b238 100644 --- a/docker/Dockerfile.gpu +++ b/docker/Dockerfile.gpu @@ -10,7 +10,7 @@ RUN pip install --upgrade pip WORKDIR /app COPY ./requirements.txt /app -RUN pip install -r /app/requirements.txt +RUN pip install -r /app/requirements.gpu.txt # Hack to get around tensorflow-io issue - https://github.com/tensorflow/io/issues/1755 RUN pip install tensorflow-io diff --git a/requirements.gpu.txt b/requirements.gpu.txt new file mode 100644 index 0000000..2a343bd --- /dev/null +++ b/requirements.gpu.txt @@ -0,0 +1,17 @@ +matplotlib==3.6.3 +jupyterlab==3.5.3 +scipy==1.10.0 +scikit-learn==1.2.1 +scikit-image==0.19.3 +pandas==1.5.3 +music21==8.1.0 +black[jupyter]==23.1.0 +click==8.0.4 +flake8==5.0.4 +kaggle==1.5.12 +pydot==1.4.2 +ipywidgets==8.0.4 +tensorflow==2.10.1 +tensorboard==2.10.1 +tensorflow_probability==0.18.0 +flake8-nb==0.5.2 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 2a343bd..cc4d4c3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ flake8==5.0.4 kaggle==1.5.12 pydot==1.4.2 ipywidgets==8.0.4 -tensorflow==2.10.1 +tensorflow-cpu==2.10.1 tensorboard==2.10.1 tensorflow_probability==0.18.0 flake8-nb==0.5.2 \ No newline at end of file