Fix #3088: Use TensorFlow 1.15.4 with CUDNN fix
This commit is contained in:
parent
34a62bd1d1
commit
16165f3ddc
|
@ -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,
|
||||
|
|
|
@ -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>`.
|
||||
|
||||
|
|
Loading…
Reference in New Issue