> srun --pty easy_ngc --help
usage: easy_ngc_impl [-h] [--cmd CMD] [--version VERSION] [--modules MODULES] [--packages PACKAGES] [--train TRAIN] [--jupyter] [--public_ssh_key PUBLIC_SSH_KEY] [--private_ssh_key PRIVATE_SSH_KEY] [--ssh_key_prefix SSH_KEY_PREFIX] [--dry] [--overlay OVERLAY] framework gpu_indices [gpu_indices ...] Run command in NVIDIA NGC container positional arguments: framework Framework to run (Tensorflow, PyTorch, etc) gpu_indices Index of GPU(s) to use optional arguments: -h, --help show this help message and exit --cmd CMD, -c CMD Command to run in shell inside container --version VERSION, -v VERSION Container version --modules MODULES Python modules to install with pip --packages PACKAGES APT Packages to install with apt-get --train TRAIN Location and parameters of training script --jupyter Run Jupyter notebook --public_ssh_key PUBLIC_SSH_KEY Location of SSH public key --private_ssh_key PRIVATE_SSH_KEY Location of SSH private key --ssh_key_prefix SSH_KEY_PREFIX Use this value for private key, and add .pub for public key --dry Run Jupyter notebook --overlay OVERLAY Use first available overlay network