Fix #3088: Use TensorFlow 1.15.4 with CUDNN fix

This commit is contained in:
Alexandre Lissy 2020-09-25 14:11:06 +02:00
parent 34a62bd1d1
commit 16165f3ddc
3 changed files with 5 additions and 2 deletions

View File

@ -43,6 +43,9 @@ RUN pip3 install --upgrade native_client/ctcdecode/dist/*.whl
# Prepare deps
RUN pip3 install --upgrade pip==20.2.2 wheel==0.34.2 setuptools==49.6.0
# FIXME: Hack because no 1.15.4 Docker image
RUN pip3 install --upgrade tensorflow-gpu==1.15.4
# Install DeepSpeech
# - No need for the decoder since we did it earlier
# - There is already correct TensorFlow GPU installed on the base image,

View File

@ -67,7 +67,7 @@ If you have a capable (NVIDIA, at least 8GB of VRAM) GPU, it is highly recommend
.. code-block:: bash
pip3 uninstall tensorflow
pip3 install 'tensorflow-gpu==1.15.2'
pip3 install 'tensorflow-gpu==1.15.4'
Please ensure you have the required :ref:`CUDA dependency <cuda-deps>`.

View File

@ -74,7 +74,7 @@ def main():
]
tensorflow_pypi_dep = [
'tensorflow == 1.15.2'
'tensorflow == 1.15.4'
]
# Due to pip craziness environment variables are the only consistent way to