Skip to content

Debugging using with ssh for Github Actions

Huy Do edited this page Dec 13, 2024 · 25 revisions

SSH for Github Actions

Platform availability:

Conditions and Limitations

  • The job must still be running when you ssh into the machine. If the job is finished running, you cannot ssh in.
  • You need to add no-runner-experiments on the PR to fallback to Meta AWS runners to be able to use ssh. With the migration to Linux Foundation firmly in place, Linux Foundation runners don't allow ssh access anymore for security reason.
  • Only works on your own PRs. If you want to ssh into a job running on the main or someone else's job, see the below section about aws ssm.
  • You will be removed from the machine after 2 hours.
  • Only works for users who are connected to the Meta VPN
    • No current planned support for outside collaborators
  • Public keys for ssh are pulled from Github using https://github.com/${github.actor}.keys

Workflow for users

  1. Push a new commit / re-run completed workflows, see below for re-running jobs through the Github UI
    image or image
    Unfortunately, jobs can only be rerun after all jobs in the workflow are finished.
  2. Traverse to logs for a build or test job that runs the Setup SSH step added (currently all of our linux workflows have this enabled)
    image
  3. Use the SSH command provided to log into the node (do this immediately, as the job will start cleaning up if it reaches the end without any active SSH session):
    Screen Shot 2021-08-04 at 11 49 25 AM

Notes for users

General

  • The default timeout for these jobs is 2 hours after workflows have completed, but only if you SSH in before the end of the job
    • Users will be kicked after workflows have either timed out or have been cancelled

VSCode

VSCode's remote SSH extension works with both Linux and Windows CI machines. Add them as you would any other SSH remote, with cmd + shift + P > SSH: New Remote and use the address runneruser@https://ec2....compute-1.amazonaws.com as provided to you in the GitHub CI logs. On Windows you can directly open VSCode to the pytorch folder with the command:

[your mac] $ export HOST=runneruser@ec2-3-238-198-8.compute-1.amazonaws.com
[your mac] $ code --folder-uri vscode-remote://ssh-remote+$HOST/c:\\actions-runner\\_work\\pytorch\\pytorch

Linux

Once you are connected through ssh, you may need to enter a docker container. Run docker ps to check if there are any docker containers running. Note that your CI job might be in the process of initiating a docker container, which means it will not show up yet. It is best to wait until the CI job reaches a step where it is building pytorch or running pytorch tests. If the job does have a docker container, run docker exec -it CONTAINER_ID /bin/bash to connect to it (or simply run docker exec -it $(docker ps | awk 'NR==2 {print $1}') /bin/bash directly to parse the CONTAINER_ID).

Now you can find the pytorch working directory, which could be ~/workspace or ~/project, and run commands locally to debug the failure.

Windows

  • To set-up the same CI environment run: C:\actions-runner\_work\pytorch\pytorch\build\win_tmp\ci_scripts\pytorch_env_restore.bat
  • The Windows workspace is currently located at C:\actions-runner\_work\pytorch\pytorch
  • To use other shells for Windows just append the shell you'd like to run to your ssh command like:
    • ssh runneruser@ec2-3-238-136-38.compute-1.amazonaws.com -- bash.exe
  • To run a particular test use the Miniconda Python: C:\jenkins\miniconda3\python.exe test_profiler.py

RDP

For certain Windows failures, it may be useful to have a full Remote Desktop connection. To use remote desktop, get a RDP client (Mac App Store) and:

  1. The Windows EC2 machines only allow SSH connections on port 22, so make a tunnel for the RDP port (default 3389) to your local machine and set a password:
# change the EC2 hostname to the one given to you in the GitHub Actions logs
[your mac] $ ssh -L 3389:localhost:3389 runneruser@ec2-3-238-198-8.compute-1.amazonaws.com

# set a password for the 'runneruser' account (it must have upper
# and lowercase letters, a number, and be at least 8 characters)
[windows] $ net user runneruser <some password>
  1. Open the RDP client, add a new PC with the hostname localhost

  2. Connect to it by double clicking, use the username runneruser with the password you just created. You should see the remote desktop window open.

(For META Employees) Debugging using AWS SSM

AWS SSM can be used to log into any currently running EC2 instances whether or not your SSH key has been added to the particular instance.

This can be used to debug currently running jobs on PRs or trunk.

Pre-requisites

  • Network connection on META VPN
  • Cloud SSO Access (bunnylol cloud fbossci)
  • Session manager plugin for AWS CLI (aws ssm start-session will fail silently without it)

Usage

aws ssm start-session --target "<instance_id>" --region "<region>"

Example:

aws ssm start-session --target i-0099d5a07d34e8904 --region us-east-1

Caveats

  • Nodes will still be reaped on their regular schedule
  • Jobs will not wait for your session to finish before concluding
  • There is no extra hold on the machine and the 2 hour timeout does not apply here, sessions will most likely end immediately at the next reap cycle
Clone this wiki locally