The milatools package provides the
mila command, which is meant to help with connecting to and interacting with the Mila cluster.
Requires Python >= 3.8
pip install git+git://github.com/mila-iqia/milatools.git
milatools, start with
Set up your access to the mila cluster interactively. Have your username and password ready!
- Set up your SSH config for easy connection with
- Set up your public key if you don't already have them
- Copy your public key over to the cluster for passwordless auth
- Set up a public key on the login node to enable ssh into compute nodes
Connect a VSCode instance to a compute node.
mila code first allocates a compute node using slurm (you can pass slurm options as well), and then calls the
code command with the appropriate options to start a remote coding session on the allocated node.
You can simply Ctrl+C the process to end the session.
usage: mila code [-h] PATH ...
-h, --help show this help message and exit
mila code path/to/my/experiment