llama-3.1-nemotron-nano-vl-8b-v1

Generate API Key

Pull and Run the NIM

Copy Code
$ docker login nvcr.io
Username: $oauthtoken
Password: <PASTE_API_KEY_HERE>

Pull and run the NVIDIA NIM with the command below. This will download the optimized model for your infrastructure.

Copy Code
export NGC_API_KEY=<PASTE_API_KEY_HERE>
export LOCAL_NIM_CACHE=~/.cache/nim
mkdir -p "$LOCAL_NIM_CACHE"
docker run -it --rm \
    --gpus all \
    --shm-size=16GB \
    -e NGC_API_KEY \
    -v "$LOCAL_NIM_CACHE:/opt/nim/.cache" \
    -u $(id -u) \
    -p 8000:8000 \
    nvcr.io/nim/nvidia/llama-3.1-nemotron-nano-vl-8b-v1:latest

Test the NIM

You can now make a local API call using this curl command:

Copy Code
curl -X 'POST' \
'http://0.0.0.0:8000/v1/chat/completions' \
-H 'accept: application/json' \
-H 'Content-Type: application/json' \
-d '{
    "temperature": 0.0,
    "top_p": 1.0,
    "model": "nvidia/llama-3.1-nemotron-nano-vl-8b-v1",
    "messages": [
        {
            "role": "user",
            "content": [
                { "type": "image_url", "image_url": { "url": "https://assets.ngc.nvidia.com/products/api-catalog/llama-cosmos-nemotron-8b-instruct/performance.png" } },
                { "type": "text", "text": "For MOE Switch XXL training what is speed of H100 over A100 and H100 with NVLink over A100?" }
            ]
        }
    ]
}'

For more details on getting started with this NIM, visit the NVIDIA NIM Docs.