#!/bin/bash

# Check if required CUDA device is available
if ! command -v nvidia-smi &> /dev/null; then
    echo "Error: NVIDIA GPU is required but not found"
    exit 1
fi

# Execute the command passed to the container
exec "$@"