RHEL 8 - Fix PIP package (#1371)

* RHEL 8.X - Fix for NumPy * NumPy - Updates
ROCm · Jun 3, 2024 · 2a7d1a5 · 2a7d1a5
1 parent 1c8c6de
commit 2a7d1a5
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/MIVisionX-setup.py b/MIVisionX-setup.py
@@ -261,10 +261,14 @@ def ERROR_CHECK(call):
     'onnx==1.12.0'
 ]
 
-pip3InferencePackagesSLES = [
+pipNumpyVersion = "numpy==1.23.0"
+with open('/etc/os-release') as f:
+    if 'VERSION_ID="8' in f.read():
+        pipNumpyVersion = "numpy==1.19.5"
+pip3InferencePackagesRPM = [
     'future==0.18.2',
     'pytz==2022.1',
-    'numpy==1.23.0',
+    str(pipNumpyVersion),
     'google==3.0.0',
     'protobuf==3.12.4',
     'onnx==1.11.0'
@@ -417,8 +421,8 @@ def ERROR_CHECK(call):
                         ERROR_CHECK(os.system('sudo '+linuxFlag+' '+linuxSystemInstall +
                                 ' '+linuxSystemInstall_check+' install -y '+ inferenceRPMPackages[i]))
                     # Install base Deps
-                    for i in range(len(pip3InferencePackagesSLES)):
-                            ERROR_CHECK(os.system('sudo pip3 install '+ pip3InferencePackagesSLES[i]))
+                    for i in range(len(pip3InferencePackagesRPM)):
+                            ERROR_CHECK(os.system('sudo pip3 install '+ pip3InferencePackagesRPM[i]))
                 # Install NNEF Deps
                 ERROR_CHECK(os.system('mkdir -p '+modelCompilerDeps+'/nnef-deps'))
                 ERROR_CHECK(os.system(

diff --git a/tests/neural_network_tests/runNeuralNetworkTests.py b/tests/neural_network_tests/runNeuralNetworkTests.py
@@ -137,10 +137,14 @@ def script_info():
     'onnx==1.12.0'
 ]
 
+pipNumpyVersion = "numpy==1.23.0"
+with open('/etc/os-release') as f:
+    if 'VERSION_ID="8' in f.read():
+        pipNumpyVersion = "numpy==1.19.5"
 pip3InferencePackagesRPM = [
     'future==0.18.2',
     'pytz==2022.1',
-    'numpy==1.23.0',
+    str(pipNumpyVersion),
     'google==3.0.0',
     'protobuf==3.12.4',
     'onnx==1.11.0'