Manual easy_ngc

easy_ngc

	> 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