From 8e63dcdb1592518b0145ae495b05be52e1dafa26 Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Fri, 3 May 2024 21:49:41 -0700 Subject: [PATCH 1/2] docker: downgrade to allow cuda 11 --- install/docker/install-nvidia-graphics.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/install/docker/install-nvidia-graphics.sh b/install/docker/install-nvidia-graphics.sh index f35eef2c0..aa39f827e 100644 --- a/install/docker/install-nvidia-graphics.sh +++ b/install/docker/install-nvidia-graphics.sh @@ -1,6 +1,9 @@ if [ "$(uname -m)" = "x86_64" ] then - export CUDA_VERSION=12-4 + # uncomment to require cuda 12, but everything currently targets 11. + # stuff that targets 11 will work with 12 though. + # export CUDA_VERSION=12-4 + export CUDA_VERSION=11-8 echo "Installing NVIDIA graphics packages." apt update -q \ && apt install wget \ From 905a9aec21c6fb4dc8ad9855c242acc47a8dc65e Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Fri, 3 May 2024 21:58:35 -0700 Subject: [PATCH 2/2] python plugins: fix linux platform check --- plugins/opencv/src/requirements.txt | 14 +++++++++----- plugins/openvino/src/requirements.txt | 4 ++-- plugins/python-codecs/src/requirements.txt | 4 ++-- plugins/tensorflow-lite/src/requirements.txt | 7 ++++--- 4 files changed, 17 insertions(+), 12 deletions(-) diff --git a/plugins/opencv/src/requirements.txt b/plugins/opencv/src/requirements.txt index 896e8c343..05185c63f 100644 --- a/plugins/opencv/src/requirements.txt +++ b/plugins/opencv/src/requirements.txt @@ -1,9 +1,13 @@ # plugin numpy>=1.16.2 -# pillow for anything not intel linux -Pillow>=5.4.1; sys_platform != 'linux' or platform_machine != 'x86_64' -pillow-simd; sys_platform == 'linux' and platform_machine == 'x86_64' + imutils>=0.5.0 -# opencv-python is not available on armhf + # locked to version because 4.8.0.76 is broken. -opencv-python==4.8.0.74; sys_platform != 'linux' or platform_machine == 'x86_64' or platform_machine == 'aarch64' +# todo: check newer versions. +opencv-python==4.8.0.74 + +# pillow-simd is available on x64 linux +# pillow-simd confirmed not building with arm64 linux or apple silicon +Pillow>=5.4.1; 'linux' not in sys_platform or platform_machine != 'x86_64' +pillow-simd; 'linux' in sys_platform and platform_machine == 'x86_64' diff --git a/plugins/openvino/src/requirements.txt b/plugins/openvino/src/requirements.txt index 9a043c50e..93e32ce59 100644 --- a/plugins/openvino/src/requirements.txt +++ b/plugins/openvino/src/requirements.txt @@ -3,5 +3,5 @@ openvino==2024.0.0 # pillow-simd is available on x64 linux # pillow-simd confirmed not building with arm64 linux or apple silicon -Pillow>=5.4.1; sys_platform != 'linux' or platform_machine != 'x86_64' -pillow-simd; sys_platform == 'linux' and platform_machine == 'x86_64' +Pillow>=5.4.1; 'linux' not in sys_platform or platform_machine != 'x86_64' +pillow-simd; 'linux' in sys_platform and platform_machine == 'x86_64' diff --git a/plugins/python-codecs/src/requirements.txt b/plugins/python-codecs/src/requirements.txt index 00ba2aeaf..35484d9bb 100644 --- a/plugins/python-codecs/src/requirements.txt +++ b/plugins/python-codecs/src/requirements.txt @@ -5,5 +5,5 @@ av>=10.0.0 # in case pyvips fails to load, use a pillow fallback. # pillow for anything not intel linux, pillow-simd is available on x64 linux -Pillow>=5.4.1; sys_platform != 'linux' or platform_machine != 'x86_64' -pillow-simd; sys_platform == 'linux' and platform_machine == 'x86_64' +Pillow>=5.4.1; 'linux' not in sys_platform or platform_machine != 'x86_64' +pillow-simd; 'linux' in sys_platform and platform_machine == 'x86_64' diff --git a/plugins/tensorflow-lite/src/requirements.txt b/plugins/tensorflow-lite/src/requirements.txt index 0700565a2..2a44d6bff 100644 --- a/plugins/tensorflow-lite/src/requirements.txt +++ b/plugins/tensorflow-lite/src/requirements.txt @@ -2,6 +2,7 @@ pycoral~=2.0 tflite-runtime==2.5.0.post1 -# pillow for anything not intel linux, pillow-simd is available on x64 linux -Pillow>=5.4.1; sys_platform != 'linux' or platform_machine != 'x86_64' -pillow-simd; sys_platform == 'linux' and platform_machine == 'x86_64' +# pillow-simd is available on x64 linux +# pillow-simd confirmed not building with arm64 linux or apple silicon +Pillow>=5.4.1; 'linux' not in sys_platform or platform_machine != 'x86_64' +pillow-simd; 'linux' in sys_platform and platform_machine == 'x86_64'