зеркало из https://github.com/mozilla/DeepSpeech.git
Fix #3088: Use TensorFlow 1.15.4 with CUDNN fix
This commit is contained in:
Родитель
34a62bd1d1
Коммит
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>`.
|
||||
|
||||
|
|
2
setup.py
2
setup.py
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче