SSH Flexibility for SageMaker clusters #105

jlewitt1 · 2023-09-10T19:01:48Z

Improve the flexibility for connecting to SageMaker clusters by:

Maintaining an authorized keys list with public keys stored in S3
Copying the authorized keys file onto the cluster when connecting
Generating a fresh key pair and adding the public key to the authorized list If user with the relevant SageMaker role does not already have one

…ster when creating the SSM session

…aram

jlewitt1 marked this pull request as draft September 10, 2023 19:01

jlewitt1 changed the title ~~SSH Flexibility for SageMaker clusters~~ [WIP] SSH Flexibility for SageMaker clusters Sep 10, 2023

jlewitt1 force-pushed the sagemaker-ssh-flexibility branch from 83dc977 to 22f9b25 Compare September 11, 2023 03:05

maintain list of all authorized public keys to be copied onto the clu…

dd3eecc

…ster when creating the SSM session

jlewitt1 force-pushed the sagemaker-ssh-flexibility branch from 22f9b25 to dd3eecc Compare September 11, 2023 03:24

SSH key generation flow cleanup. add region as constructor / config p…

504a411

…aram

jlewitt1 changed the title ~~[WIP] SSH Flexibility for SageMaker clusters~~ SSH Flexibility for SageMaker clusters Sep 13, 2023

jlewitt1 marked this pull request as ready for review September 13, 2023 18:40

jlewitt1 force-pushed the sagemaker-ssh-flexibility branch 3 times, most recently from 0393dd8 to 86ff7d1 Compare September 14, 2023 20:31

simplify SSH key management - create a new keypair if not found locally

ae7c48d

jlewitt1 force-pushed the sagemaker-ssh-flexibility branch from 86ff7d1 to ae7c48d Compare September 16, 2023 01:00

jlewitt1 merged commit 85fcef0 into main Sep 16, 2023

jlewitt1 deleted the sagemaker-ssh-flexibility branch September 29, 2023 22:44

Provide feedback