gpu-container-setup-flagos

Name: gpu-container-setup-flagos
Availability: InStock
Author: FlagOS

Detect multi-vendor GPU environments and launch compatible PyTorch containers.

18|12|Updated Mar 3, 2026

Search Tags:#docker #gpu #container #pytorch #nvidia #gpu-detection #ascend

Ready to run

npx skills add https://github.com/flagos-ai/skills --skill gpu-container-setup-flagos

💬 Prompt for ChatGPT / Claude Chat▼

Please help me install this Agent Skill.
Skill: gpu-container-setup-flagos
Source: https://github.com/flagos-ai/skills/tree/main/skills/gpu-container-setup-flagos
Command: npx skills add https://github.com/flagos-ai/skills --skill gpu-container-setup-flagos

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) and references (resource) components.

What problem does it solve?

Automatically detects multi-vendor GPU environments, locates a matching PyTorch container image, and prepares the host mounts to run and validate GPU workloads.

Core Features & Use Cases

Detects NVIDIA, AMD/ROCm, Ascend, Metax, and Iluvatar GPUs
Discovers and selects PyTorch container images from primary vendor hubs, Harbor, and fallback sources
Validates GPU availability inside the container using a runtime validation script

Quick Start

To begin, ask the agent to run the GPU container setup to automatically detect your GPU vendor and launch the correct PyTorch container.

Frequently Asked Questions about gpu-container-setup-flagos

High-intent search queries and answers about installing and using this skill.

FAQPage Schema

How do I automatically detect my GPU type and start a PyTorch container?▼

To automatically detect your GPU type and start a PyTorch container, this skill identifies your hardware environment, selects a matching image from vendor hubs, prepares host data mounts, and runs a runtime validation script to verify GPU availability inside the container.

Does this PyTorch container setup support Ascend and Metax GPUs?▼

Yes, this PyTorch container setup supports Ascend and Metax GPUs. It is designed for multi-vendor GPU support, automatically detecting NVIDIA, AMD/ROCm, Ascend, Metax, and Iluvatar environments to locate and launch compatible workloads.

What's the best way to run a PyTorch Docker container across different GPU vendors?▼

The best way to run a PyTorch Docker container across different GPU vendors is to use an automated detection and selection logic that discovers compatible images from primary vendor hubs, Harbor, and fallback sources, ensuring hardware compatibility before execution.

How does GPU validation work inside a container after it launches?▼

GPU validation inside a container works by running a runtime validation script after launch. This script checks the prepared host data mounts and confirms that the detected GPU hardware is successfully mapped and accessible for PyTorch workloads.

Do I need to manually specify a container image registry for my GPU environment?▼

No, you do not need to manually specify a container image registry for your GPU environment. The setup discovers and selects PyTorch images automatically by checking primary vendor hubs, Harbor, and fallback sources based on the detected hardware.

Why does my GPU container setup fail to recognize multi-vendor hardware?▼

Your GPU container setup might fail to recognize multi-vendor hardware if the detection logic cannot locate a matching image. The skill mitigates this by validating GPU availability inside the container and using fallback sources if primary vendor hubs lack the required image.