SSH configuration for AI Workbench running remotely on brev.dev

AI & Data Science NVIDIA AI Workbench
1

I’m trying to run AI Workbench remotely on a brev.dev VM. brev.dev lets me rent a GPU and I’d like to try some of my models on larger GPUs.

The steps are

  1. Spin up a brev.dev VM usually named awesome-gpu-name
  2. Download and run the brev CLI on my Mac
  3. Run the brev.dev login to generate credentials
  4. Run brev shell to set up an SSH connection and the login credentials. Now I’m logged into my brev.dev VM awesome-gpu-name-host on an ssh tunnel.
  5. Exit the brev shell
  6. Try ssh awesome-gpu-name-host which sets up the fingerprints blah blah and does an ssh into awesome-gpu-name-host
  7. So now I have the ability to set up an SSH connection between my Mac and the brev.dev VM
  8. Open AI Workbench
  9. Click on add a remote location
  10. I can enter the IP address associated with that awesome-gpu-name entry that I have in my .brev/ssh_config that I used to verify above?

What do I enter into the SSH Key File field. brev.dev uses a pem file. I can’t seem to figure out what file goes in this field.

Please tick the appropriate box to help us categorize your post
Bug or Error
Feature Request
Documentation Issue
Other