[Azure][Autoscaler] Allow current user to use Docker (#15380)

python/ray/autoscaler/azure/defaults.yaml

@@ -94,7 +94,7 @@ head_node_type: ray.head.default
 file_mounts: {
 #    "/path1/on/remote/machine": "/path1/on/local/machine",
 #    "/path2/on/remote/machine": "/path2/on/local/machine",
-     "/home/ubuntu/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
+     "~/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
 }
 
 # Files or directories to copy from the head node to the worker nodes. The format is a
@@ -120,6 +120,7 @@ rsync_filter: []
 # is setup.
 initialization_commands:
     # get rid of annoying Ubuntu message
+    - sudo usermod -aG docker $USER || true
     - touch ~/.sudo_as_admin_successful
 
 # List of shell commands to run to set up nodes.

python/ray/autoscaler/azure/example-full-legacy.yaml

@@ -90,7 +90,7 @@ worker_nodes:
 file_mounts: {
 #    "/path1/on/remote/machine": "/path1/on/local/machine",
 #    "/path2/on/remote/machine": "/path2/on/local/machine",
-     "/home/ubuntu/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
+     "~/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
 }
 
 # Files or directories to copy from the head node to the worker nodes. The format is a
@@ -119,6 +119,7 @@ rsync_filter:
 # is setup.
 initialization_commands:
     # get rid of annoying Ubuntu message
+    - sudo usermod -aG docker $USER || true
     - touch ~/.sudo_as_admin_successful
 
 # List of shell commands to run to set up nodes.

python/ray/autoscaler/azure/example-full.yaml

@@ -111,7 +111,7 @@ head_node_type: ray.head.default
 file_mounts: {
 #    "/path1/on/remote/machine": "/path1/on/local/machine",
 #    "/path2/on/remote/machine": "/path2/on/local/machine",
-     "/home/ubuntu/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
+     "~/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
 }
 
 # Files or directories to copy from the head node to the worker nodes. The format is a
@@ -140,6 +140,7 @@ rsync_filter:
 # is setup.
 initialization_commands:
     # get rid of annoying Ubuntu message
+    - sudo usermod -aG docker $USER || true
     - touch ~/.sudo_as_admin_successful
 
 # List of shell commands to run to set up nodes.

python/ray/autoscaler/azure/example-gpu-docker.yaml

@@ -81,9 +81,17 @@ head_node_type: ray.head.gpu
 file_mounts: {
 #    "/path1/on/remote/machine": "/path1/on/local/machine",
 #    "/path2/on/remote/machine": "/path2/on/local/machine",
-     "/home/ubuntu/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
+     "~/.ssh/id_rsa.pub": "~/.ssh/id_rsa.pub"
 }
 
+# List of commands that will be run before `setup_commands`. If docker is
+# enabled, these commands will run outside the container and before docker
+# is setup.
+initialization_commands:
+    # get rid of annoying Ubuntu message
+    - sudo usermod -aG docker $USER || true
+    - touch ~/.sudo_as_admin_successful
+
 # List of shell commands to run to set up nodes.
 # NOTE: rayproject/ray-ml:latest has ray latest bundled
 setup_commands: []