diff --git a/docs/gpu-k8s-role.md b/docs/gpu-k8s-role.md index 3f40606..6c5853b 100644 --- a/docs/gpu-k8s-role.md +++ b/docs/gpu-k8s-role.md @@ -30,8 +30,8 @@ sealos run \ --env '{}' \ --cmd "kubeadm init --skip-phases=addon/kube-proxy" ``` -If deploying with a non-root user the command also requires `--ssh-user` and -`--ssh-private-key` options pointing to the user's key. +If deploying with a non-root user the command also requires `--user` and +`--pk` options pointing to the user's SSH key. After the cluster is running the role installs the NVIDIA device plugin and runs a test pod to ensure `nvidia-smi` works inside the cluster. diff --git a/playbooks/roles/vhosts/gpu-k8s/tasks/install_cluster.yml b/playbooks/roles/vhosts/gpu-k8s/tasks/install_cluster.yml index 10e6837..09d4096 100644 --- a/playbooks/roles/vhosts/gpu-k8s/tasks/install_cluster.yml +++ b/playbooks/roles/vhosts/gpu-k8s/tasks/install_cluster.yml @@ -95,8 +95,8 @@ {{ labring_registry.stdout }}/helm:{{ helm_version }} \ --masters {{ master_ips | join(',') }} \ --nodes {{ node_ips | join(',') }} \ - --ssh-user {{ ssh_user }} \ - --ssh-private-key {{ ssh_private_key }} \ + --user {{ ssh_user }} \ + --pk {{ ssh_private_key }} \ --env '{{ sealos_cmd_env }}' \ --cmd "{{ kubeadm_init_cmd }}" args: