Setup - PIP Package Updates (#1366)

Co-authored-by: Kiriti Gowda <kiritigowda@gmail.com>
ROCm · May 31, 2024 · 79bbcaa · 79bbcaa
1 parent 028b210
commit 79bbcaa
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/MIVisionX-setup.py b/MIVisionX-setup.py
@@ -252,7 +252,7 @@ def ERROR_CHECK(call):
     'python3-protobuf'
 ]
 
-pip3InferencePackages = [
+pip3InferencePackagesUbuntu = [
     'future==0.18.2',
     'pytz==2022.1',
     'numpy==1.21',
@@ -261,6 +261,15 @@ def ERROR_CHECK(call):
     'onnx==1.12.0'
 ]
 
+pip3InferencePackagesSLES = [
+    'future==0.18.2',
+    'pytz==2022.1',
+    'numpy==1.19',
+    'google==3.0.0',
+    'protobuf==3.12.4',
+    'onnx==1.11.0'
+]
+
 ffmpegDebianPackages = [
     'ffmpeg',
     'libavcodec-dev',
@@ -400,13 +409,16 @@ def ERROR_CHECK(call):
                     for i in range(len(inferenceDebianPackages)):
                         ERROR_CHECK(os.system('sudo '+linuxFlag+' '+linuxSystemInstall +
                                 ' '+linuxSystemInstall_check+' install -y '+ inferenceDebianPackages[i]))
+                    # Install base Deps
+                    for i in range(len(pip3InferencePackagesUbuntu)):
+                        ERROR_CHECK(os.system('sudo pip3 install '+ pip3InferencePackagesUbuntu[i]))
                 else:
                     for i in range(len(inferenceRPMPackages)):
                         ERROR_CHECK(os.system('sudo '+linuxFlag+' '+linuxSystemInstall +
                                 ' '+linuxSystemInstall_check+' install -y '+ inferenceRPMPackages[i]))
-                # Install base Deps
-                for i in range(len(pip3InferencePackages)):
-                        ERROR_CHECK(os.system('sudo pip3 install '+ pip3InferencePackages[i]))
+                    # Install base Deps
+                    for i in range(len(pip3InferencePackagesSLES)):
+                            ERROR_CHECK(os.system('sudo pip3 install '+ pip3InferencePackagesSLES[i]))
                 # Install NNEF Deps
                 ERROR_CHECK(os.system('mkdir -p '+modelCompilerDeps+'/nnef-deps'))
                 ERROR_CHECK(os.system(