Support for active ssh connections

[albanie]

Do you think it would be possible to support the use-case where the user logs in to a "login node" (with no GPUs) and then connects from there to a compute node (which contains the GPUs) via ssh using the integrated terminal? Would it be feasible to view the GPU usage of a currently active ssh connection, using the same visualisation (e.g. by making repeated remote calls to nvidia-smi)?

[innerlee]

If you remotely connected to the login node, surely it is possible to implement.

Step 1: write a script that queries gpu usage (not related to extension)
Step 2: add an extension config to parse the output of user provided script (pr welcome)

[mingjun97]

You may ues the remote-ssh plugin to attach current workspace to the remote machine via ssh, then install nvidia-smi as you did in local machine. The remote-ssh also enables you to edit files directly on the remote machine.

This works pertty cool!