Install aarch pytorch package if linux os

tiiuae · Nov 16, 2023 · ab85110 · ab85110
1 parent 223ef51
commit ab85110
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/modules/utils/docker/entrypoint_nats.sh b/modules/utils/docker/entrypoint_nats.sh
@@ -74,9 +74,12 @@ else
   /opt/S90comms_controller start
 
   # Install python package "torch"
-  echo "installing packages"
-  chmod +x /opt/mesh_com/modules/utils/package/install_dependencies.sh
-  /opt/mesh_com/modules/utils/package/install_dependencies.sh
+  os_type=$(uname -s)
+  if [ "$os_type" = "Linux" ]; then
+      echo "installing packages"
+      chmod +x /opt/mesh_com/modules/utils/package/install_dependencies.sh
+      /opt/mesh_com/modules/utils/package/install_dependencies.sh
+  fi
 
   # alive
   nohup /bin/bash -c "while true; do sleep infinity; done"