updated pytorch installation for aarch64

tiiuae · Nov 17, 2023 · bd6f4b7 · bd6f4b7
1 parent ab85110
commit bd6f4b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/utils/docker/entrypoint_nats.sh b/modules/utils/docker/entrypoint_nats.sh
@@ -74,8 +74,8 @@ else
   /opt/S90comms_controller start
 
   # Install python package "torch"
-  os_type=$(uname -s)
-  if [ "$os_type" = "Linux" ]; then
+  machine_hardware=$(uname -m)
+  if [ "machine_hardware" = "aarch64" ]; then
       echo "installing packages"
       chmod +x /opt/mesh_com/modules/utils/package/install_dependencies.sh
       /opt/mesh_com/modules/utils/package/install_dependencies.sh