Unable to build diffusion containers after recent updates to opencv

[ms1design]

Hi @dusty-nv,

Just taking a note that we had a breaking change in opencv that negatively impacted the diffusion containers breaking the builds:

In opencv we have new way of install/building:

jetson-containers/packages/opencv/Dockerfile

Lines 25 to 29 in 487dc4f

	RUN cd /tmp/opencv && \
	./install_deps.sh && \
	./install_pip.sh || \
	./install_deb.sh || \
	./build.sh

But this fails hard on stable-diffusion container where invisible-watermark dependency requires opencv-python and cannot find proper distribution.

Same with stable-diffusion-webui:

jetson-containers/packages/diffusion/stable-diffusion-webui/Dockerfile

Lines 35 to 36 in 487dc4f

	# partially initialized module 'cv2' has no attribute 'gapi_wip_gst_GStreamerPipeline' \
	&& /opt/opencv_install.sh \

10:44:49 INFO: pip is looking at multiple versions of facexlib to determine which version is compatible with other requirements. This could take a while.
10:44:49 
10:44:49 stderr: ERROR: Ignored the following versions that require a different python version: 0.52.0 Requires-Python >=3.6,<3.9; 0.52.0rc3 Requires-Python >=3.6,<3.9; 0.53.0 Requires-Python >=3.6,<3.10; 0.53.0rc1.post1 Requires-Python >=3.6,<3.10; 0.53.0rc2 Requires-Python >=3.6,<3.10; 0.53.0rc3 Requires-Python >=3.6,<3.10; 0.53.1 Requires-Python >=3.6,<3.10; 0.54.0 Requires-Python >=3.7,<3.10; 0.54.0rc2 Requires-Python >=3.7,<3.10; 0.54.0rc3 Requires-Python >=3.7,<3.10; 0.54.1 Requires-Python >=3.7,<3.10
10:44:49 ERROR: Could not find a version that satisfies the requirement opencv-python (from facexlib) (from versions: none)
10:44:49 ERROR: No matching distribution found for opencv-python

Started to fix that in hope it's a quick one – running the opencv install/build sh scripts, but I gave up when realised that it starts the build.sh from opencv. build.sh requires the unavailable environment variables (accessible only when building opencv cotnainer):

jetson-containers/packages/opencv/config.py

Lines 9 to 12 in 487dc4f

	cv['build_args'] = {
	'OPENCV_VERSION': version,
	'OPENCV_PYTHON': f"{version.split('.')[0]}.x",
	'CUDA_ARCH_BIN': ','.join([f'{x/10:.1f}' for x in CUDA_ARCHITECTURES]),

[ai-and-i]

Hi, I have the same issue. @ms1design where you able to solve it or work around it?

[ms1design]

Unfortunately, no, but I haven't tried yet tbh. I have this on my to-do list for when I have more time available.

[ai-and-i]

I think I fixed it in #541

[dusty-nv]

Thanks @ai-and-i , took a deeper look and fixed it another way in e0368bf

It was related to the CUDA-enabled opencv wheels being called opencv-contrib-python not opencv-python, but yet I was blacklisting opencv-python on the pip server from mirroring PyPi (so it wouldn't install the non-CUDA versions). So I disabled that one since it was creating errors during pip installs, and instead reinstalled the OpenCV-CUDA binaries near the end of the build.

[ai-and-i]

Perfect, thanks for the quick fix!

[ms1design]

Fixed in e0368bf, closing.