gpu-worker 1.0.0

package/scripts/postinstall.js

@@ -0,0 +1,60 @@
+#!/usr/bin/env node
+/**
+ * Post-install script
+ * 在 npm install 后自动设置 Python 环境
+ */
+
+const { execSync } = require('child_process');
+const path = require('path');
+const fs = require('fs');
+
+const PACKAGE_DIR = path.resolve(__dirname, '..');
+
+// ANSI colors (不依赖 chalk)
+const colors = {
+    green: (s) => `\x1b[32m${s}\x1b[0m`,
+    yellow: (s) => `\x1b[33m${s}\x1b[0m`,
+    cyan: (s) => `\x1b[36m${s}\x1b[0m`,
+    red: (s) => `\x1b[31m${s}\x1b[0m`,
+    bold: (s) => `\x1b[1m${s}\x1b[0m`
+};
+
+console.log(colors.cyan('\n📦 GPU Worker - Post-install setup\n'));
+
+// 检测 Python
+function findPython() {
+    const pythonCommands = ['python3', 'python', 'py'];
+
+    for (const cmd of pythonCommands) {
+        try {
+            const version = execSync(`${cmd} --version`, {
+                encoding: 'utf8',
+                stdio: ['pipe', 'pipe', 'pipe']
+            });
+            const match = version.match(/Python (\d+)\.(\d+)/);
+            if (match && parseInt(match[1]) >= 3 && parseInt(match[2]) >= 9) {
+                return { cmd, version: version.trim() };
+            }
+        } catch (e) {
+            continue;
+        }
+    }
+    return null;
+}
+
+const python = findPython();
+
+if (python) {
+    console.log(colors.green(`✓ Found ${python.version}`));
+    console.log(colors.cyan('\nRun the following to get started:'));
+    console.log(colors.bold('  npx gpu-worker configure'));
+    console.log(colors.bold('  npx gpu-worker start'));
+} else {
+    console.log(colors.yellow('⚠ Python 3.9+ not found'));
+    console.log(colors.yellow('\nPlease install Python 3.9+ before using gpu-worker:'));
+    console.log('  - Windows: https://www.python.org/downloads/');
+    console.log('  - macOS:   brew install python@3.11');
+    console.log('  - Ubuntu:  sudo apt install python3.11 python3.11-venv');
+}
+
+console.log(colors.cyan('\nFor more info: npx gpu-worker --help\n'));

package/setup.py

@@ -0,0 +1,43 @@
+#!/usr/bin/env python3
+"""
+GPU Worker 安装脚本
+支持 pip install 安装
+"""
+from setuptools import setup, find_packages
+
+setup(
+    name="gpu-worker",
+    version="1.0.0",
+    description="分布式GPU推理 Worker 节点",
+    author="GPU Sharing Project",
+    python_requires=">=3.9",
+    packages=find_packages(),
+    install_requires=[
+        "torch>=2.0.0",
+        "transformers>=4.35.0",
+        "diffusers>=0.24.0",
+        "accelerate>=0.24.0",
+        "peft>=0.6.0",
+        "httpx>=0.25.0",
+        "pyyaml>=6.0",
+        "pydantic>=2.0.0",
+        "fastapi>=0.100.0",
+        "uvicorn>=0.23.0",
+        "rich>=13.0.0",
+    ],
+    extras_require={
+        "cuda": ["bitsandbytes>=0.41.0"],
+    },
+    entry_points={
+        "console_scripts": [
+            "gpu-worker=cli:main",
+        ],
+    },
+    classifiers=[
+        "Development Status :: 4 - Beta",
+        "Intended Audience :: Developers",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+    ],
+)