diff --git a/keras3-cpu/Dockerfile b/keras3-cpu/Dockerfile index 15e0a7b6177a938495ebfb1d64c66adf30025b8d..62233222ae8ab9eb5eec62e6435e6351f6b8f74b 100644 --- a/keras3-cpu/Dockerfile +++ b/keras3-cpu/Dockerfile @@ -122,8 +122,8 @@ RUN mamba install --yes \ 'jupyter-archive' \ 'jupyterlab-git' \ 'jupyter-resource-usage' \ - # install opencv via fastai, headless version not available in conda-forge - 'fastai::opencv-python-headless' && \ + # install opencv headless version + 'py-opencv=*=headless*' && \ printf '%s\n' \ "" \ "# track CPU usage, prevent known bug with prometheus, set default limits if not set by MEM_LIMIT and CPU_LIMIT" \ diff --git a/keras3-gpu/Dockerfile b/keras3-gpu/Dockerfile index b5deea16b14d91b59f0798db027730718b100271..d3bd17a07f04aa2645b551b76d2885c79fa82a3f 100644 --- a/keras3-gpu/Dockerfile +++ b/keras3-gpu/Dockerfile @@ -74,8 +74,8 @@ RUN mamba install --yes \ 'jupyterlab_execute_time' \ 'jupyter-archive' \ 'jupyter-resource-usage' \ - # install opencv via fastai, headless version not available in conda-forge - 'fastai::opencv-python-headless' && \ + # install opencv headless version + 'py-opencv=*=headless*' && \ printf '%s\n' \ "" \ "# track CPU usage, prevent known bug with prometheus, set default limits if not set by MEM_LIMIT and CPU_LIMIT" \ diff --git a/pytorch-gpu/Dockerfile b/pytorch-gpu/Dockerfile index 5d1ff7279650e21b0d3ba6563632db989edfb409..1dbe067d5c427696903495dc684b22be47057c12 100644 --- a/pytorch-gpu/Dockerfile +++ b/pytorch-gpu/Dockerfile @@ -72,8 +72,8 @@ RUN mamba install --yes --no-update-deps \ 'jupyterlab_execute_time' \ 'jupyter-archive' \ 'jupyter-resource-usage' \ - # install opencv via fastai, headless version not available in conda-forge - 'fastai::opencv-python-headless' && \ + # install opencv headless version + 'py-opencv=*=headless*' && \ pip install --no-cache-dir --extra-index-url 'https://pypi.nvidia.com' --extra-index-url 'https://download.pytorch.org/whl/cu118' \ 'transformers' \ 'diffusers' \