diff --git a/tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu b/tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu index b73bb8df84a1..b5aa9e95e112 100644 --- a/tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu +++ b/tensorflow/training/docker/2.19/py3/cu125/Dockerfile.gpu @@ -211,7 +211,7 @@ RUN ${PIP} install --no-cache-dir -U \ mpi4py \ h5py \ absl-py \ - opencv-python \ + "opencv-python==4.11.0.86" \ werkzeug \ urllib3 \ protobuf @@ -224,6 +224,20 @@ RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_confi # Add NGC vars ENV TF_AUTOTUNE_THRESHOLD=2 +# patch nvjpeg +RUN mkdir -p /tmp/nvjpeg \ +&& cd /tmp/nvjpeg \ +&& wget https://developer.download.nvidia.com/compute/cuda/redist/libnvjpeg/linux-x86_64/libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz \ +&& tar -xvf libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz \ +&& rm -rf /usr/local/cuda/targets/x86_64-linux/lib/libnvjpeg* \ +&& rm -rf /usr/local/cuda/targets/x86_64-linux/include/nvjpeg.h \ +&& cp libnvjpeg-linux-x86_64-12.4.0.76-archive/lib/libnvjpeg* /usr/local/cuda/targets/x86_64-linux/lib/ \ +&& cp libnvjpeg-linux-x86_64-12.4.0.76-archive/include/* /usr/local/cuda/targets/x86_64-linux/include/ \ +&& rm -rf /tmp/nvjpeg \ +# patch cuobjdump and nvdisasm +&& rm -rf /usr/local/cuda/bin/cuobjdump* \ +&& rm -rf /usr/local/cuda/bin/nvdisasm* + ADD https://raw.githubusercontent.com/aws/deep-learning-containers/master/src/deep_learning_container.py /usr/local/bin/deep_learning_container.py RUN chmod +x /usr/local/bin/deep_learning_container.py @@ -338,7 +352,7 @@ RUN $PYTHON -m pip install --no-cache-dir -U \ numba \ bokeh \ imageio \ - opencv-python \ + "opencv-python==4.11.0.86" \ plotly \ seaborn \ shap