gpu-container-setup-flagos

Official

Auto-detect GPUs and launch a PyTorch container.

Authorflagos-ai
Version1.0.0
Installs0

System Documentation

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.

Dependency Matrix

Required Modules

None required

Components

scriptsreferences

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: gpu-container-setup-flagos
Download link: https://github.com/flagos-ai/skills/archive/main.zip#gpu-container-setup-flagos

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.