支持矩阵#
模型#
模型名称 |
模型 ID |
发布者 |
|---|---|---|
AI 生成图像检测 |
hive/ai-generated-image-detection |
Hive |
Deepfake 图像检测 |
hive/deepfake-image-detection |
Hive |
支持的硬件#
AI 生成图像检测#
NVIDIA 支持在以下 GPU 型号上使用优化的 TensorRT 引擎的 AI 生成图像检测模型。
GPU |
GPU 内存 (GB) |
精度 |
|---|---|---|
A100 |
40/80 |
FP32 |
H100 |
80 |
FP32 |
L40 |
48 |
FP32 |
A10G |
24 |
FP32 |
NVIDIA 支持在任何带有 ONNX 的 NVIDIA GPU 上使用 AI 生成图像检测模型。
GPU |
GPU 内存 (GB) |
精度 |
|---|---|---|
任何具有足够 GPU 内存的 NVIDIA GPU |
4 |
FP32 |
Deepfake 图像检测#
NVIDIA 支持在以下 GPU 型号上使用优化的 TensorRT 引擎的 Deepfake 图像检测模型。
GPU |
GPU 内存 (GB) |
精度 |
|---|---|---|
A100 |
40/80 |
FP32 |
H100 |
80 |
FP32 |
L40 |
48 |
FP32 |
A10G |
24 |
FP32 |
NVIDIA 支持在任何带有 ONNX 的 NVIDIA GPU 上使用 Deepfake 图像检测。
GPU |
GPU 内存 (GB) |
精度 |
|---|---|---|
任何具有足够 GPU 内存的 NVIDIA GPU |
4 |
FP32 |
软件#
NVIDIA 驱动程序#
NVIDIA NIM 多模态安全使用 Triton Inference Server Production Branch May 2024 (24.03)。 请参阅服务器的发行说明以确定驱动程序要求。
NVIDIA Container Toolkit#
您的 Docker 环境必须支持 NVIDIA GPU。 有关更多信息，请参阅安装 NVIDIA Container Toolkit。