Run Training Using Financial Fraud Training#

This document provides step-by-step instructions to log in, pull the Financial Fraud Training Docker image, run the Financial Fraud Training container, and train models using it.

1. Authenticate with NVIDIA GPU Cloud (NGC) and Pull the Docker Image#

The following explains how to authenticate with the NVIDIA GPU Cloud (NGC) and pull the Docker image from the NVIDIA registry.

Prerequisites#

  • NGC Account & API Key: Ensure you have an NGC account and a valid API key.

  • Docker: Make sure Docker is installed and running on your machine.

Authenticate with NGC#

Log in to the NVIDIA container registry using your NGC API key:

docker login nvcr.io --username '$oauthtoken' --password $NGC_API_KEY

Note: Replace $NGC_API_KEY with your actual API key or set it in NGC_API_KEY env variable.

Pull the Docker Image#

After authenticating, pull the desired Docker image from the registry:

docker pull nvcr.io/nvidia/cugraph/financial-fraud-training:2.0.0

2. Train models#

Run the Container#

Run the training container with the following command:

    docker run -it --rm --name=financial-fraud-training \
    --gpus "device=0" -v <YOUR_GNN_DATA_DIR>:/data \
    -v <DIR_TO_SAVE_TRAINED_MODELS>:/trained_models \
    nvcr.io/nvidia/cugraph/financial-fraud-training:2.0.0

Command Breakdown#

Docker Run Options:

-it: Run the container interactively.


--rm: Automatically removes the container when it exits.


--name=financial-fraud-training: Names the container.

Volume Mounts:

-v <YOUR_GNN_DATA_DIR>:/data: Mounts your GNN data directory into the container.


-v <DIR_TO_SAVE_TRAINED_MODELS>:/trained_models: Mounts a directory for saving trained models.

Note: Replace <YOUR_GNN_DATA_DIR> and <DIR_TO_SAVE_TRAINED_MODELS> with the actual paths on your system where your GNN data is located and trained models should reside.