updated pytorch installation for aarch64

tiiuae · Nov 17, 2023 · 8ed46e0 · 8ed46e0
1 parent bd6f4b7
commit 8ed46e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/utils/docker/entrypoint_nats.sh b/modules/utils/docker/entrypoint_nats.sh
@@ -75,7 +75,7 @@ else
 
   # Install python package "torch"
   machine_hardware=$(uname -m)
-  if [ "machine_hardware" = "aarch64" ]; then
+  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