How to use the extension to run on slurm compute node?

[Cloudptj]

I'm trying to use vscode to run jupyter on slurm. It's easy to run on login node. But, as you know, if you run too much on the login node, the administrator will ban your account. So, I will apply for a compute node before I run jupyter:

srun -p cpu -n 4 --pty /bin/bash

Then, the server will give a node, such as node001.

If I connect to the server with mobaxterm, I can set the node name in tunneling. Then, I can use the compute node to run jupyter in a browser.

But how to do this in vscode's jupyter extension? Is there a way to set the compute node in the extension?

I don't want to run it in a browser. And the server's compute node is not allowed to connect to the publice net.