gpu-server-setup 0.4.8__tar.gz → 0.4.9__tar.gz

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpu-server-setup
-Version: 0.4.8
+Version: 0.4.9
 Summary: Один клик — удалённый GPU-сервер с Ollama + VS Code Remote
 Author: The Fool
 License: MIT
@@ -29,6 +29,7 @@ Dynamic: license-file
 - Установка Ollama и запуск сервиса
 - Загрузка моделей с живым выводом прогресса
 - Автоматическое открытие VS Code Remote
+- Полноценный Colab-режим: Jupyter Notebook + SSH-туннель на локальный порт + venv с torch
 - Режим `--dry-run` для просмотра плана
 - Красивый вывод с помощью `rich`
 
@@ -66,14 +67,14 @@ Dynamic: license-file
 pip install gpu-server-setup
 ```
 
-После установки команда `gpu-setup` будет доступна в терминале.
+После установки команда `gpu-server-setup` будет доступна в терминале.
 
 ---
 
 ## Использование
 
 ```bash
-gpu-setup [команда] [опции]
+gpu-server-setup [команда] [опции]
 ```
 
 ### Доступные команды
@@ -84,7 +85,10 @@ gpu-setup [команда] [опции]
 | `ssh`             | Только настройка SSH-доступа                  |
 | `ollama-install`  | Установка Ollama                              |
 | `model-pull`      | Загрузка модели                               |
-| `vscode`          | Открытие VS Code Remote                       |
+| `vscode`          | Открытие VS Code 
+colab                               |
+| `colab`          | upyter Notebook + локальный туннель 
+Remote                       |
 | `plan`            | Показать план действий                        |
 
 ---
@@ -132,7 +136,7 @@ gpu-setup [команда] [опции]
 
 **Пример:**
 ```bash
-gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
+gpu-server-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
 ```
 
 **Опции:**
@@ -152,37 +156,43 @@ gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pe
 ### 1. Полная настройка со своим ключом (классический способ)
 
 ```bash
-gpu-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
+gpu-server-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
 ```
 
 ### 2. Полная настройка с `.pem`-ключом (самый удобный для облака)
 
 ```bash
-gpu-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ### 3. Только SSH с `.pem`-ключом
 
 ```bash
-gpu-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 4. Установка Ollama с `.pem`-ключом
 
 ```bash
-gpu-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 5. Скачивание модели (после настройки SSH)
 
 ```bash
-gpu-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
-### 6. Просмотр плана
+### 6. Настройка colab
 
 ```bash
-gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup colab --ip 123.45.67.89 --user ubuntu  --pem-key ~/key.pem (или --password) --port 9999
+```
+
+### 7. Просмотр плана
+
+```bash
+gpu-server-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ---
@@ -192,7 +202,7 @@ gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-u
 Все команды поддерживают `--dry-run`:
 
 ```bash
-gpu-setup setup --ip IP --user user --pem-key key.pem --dry-run
+gpu-server-setup setup --ip IP --user user --pem-key key.pem --dry-run
 ```
 
 ---

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/README.md

@@ -13,6 +13,7 @@
 - Установка Ollama и запуск сервиса
 - Загрузка моделей с живым выводом прогресса
 - Автоматическое открытие VS Code Remote
+- Полноценный Colab-режим: Jupyter Notebook + SSH-туннель на локальный порт + venv с torch
 - Режим `--dry-run` для просмотра плана
 - Красивый вывод с помощью `rich`
 
@@ -50,14 +51,14 @@
 pip install gpu-server-setup
 ```
 
-После установки команда `gpu-setup` будет доступна в терминале.
+После установки команда `gpu-server-setup` будет доступна в терминале.
 
 ---
 
 ## Использование
 
 ```bash
-gpu-setup [команда] [опции]
+gpu-server-setup [команда] [опции]
 ```
 
 ### Доступные команды
@@ -68,7 +69,10 @@ gpu-setup [команда] [опции]
 | `ssh`             | Только настройка SSH-доступа                  |
 | `ollama-install`  | Установка Ollama                              |
 | `model-pull`      | Загрузка модели                               |
-| `vscode`          | Открытие VS Code Remote                       |
+| `vscode`          | Открытие VS Code 
+colab                               |
+| `colab`          | upyter Notebook + локальный туннель 
+Remote                       |
 | `plan`            | Показать план действий                        |
 
 ---
@@ -116,7 +120,7 @@ gpu-setup [команда] [опции]
 
 **Пример:**
 ```bash
-gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
+gpu-server-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
 ```
 
 **Опции:**
@@ -136,37 +140,43 @@ gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pe
 ### 1. Полная настройка со своим ключом (классический способ)
 
 ```bash
-gpu-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
+gpu-server-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
 ```
 
 ### 2. Полная настройка с `.pem`-ключом (самый удобный для облака)
 
 ```bash
-gpu-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ### 3. Только SSH с `.pem`-ключом
 
 ```bash
-gpu-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 4. Установка Ollama с `.pem`-ключом
 
 ```bash
-gpu-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 5. Скачивание модели (после настройки SSH)
 
 ```bash
-gpu-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
-### 6. Просмотр плана
+### 6. Настройка colab
 
 ```bash
-gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup colab --ip 123.45.67.89 --user ubuntu  --pem-key ~/key.pem (или --password) --port 9999
+```
+
+### 7. Просмотр плана
+
+```bash
+gpu-server-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ---
@@ -176,7 +186,7 @@ gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-u
 Все команды поддерживают `--dry-run`:
 
 ```bash
-gpu-setup setup --ip IP --user user --pem-key key.pem --dry-run
+gpu-server-setup setup --ip IP --user user --pem-key key.pem --dry-run
 ```
 
 ---

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/gpu_server_setup.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpu-server-setup
-Version: 0.4.8
+Version: 0.4.9
 Summary: Один клик — удалённый GPU-сервер с Ollama + VS Code Remote
 Author: The Fool
 License: MIT
@@ -29,6 +29,7 @@ Dynamic: license-file
 - Установка Ollama и запуск сервиса
 - Загрузка моделей с живым выводом прогресса
 - Автоматическое открытие VS Code Remote
+- Полноценный Colab-режим: Jupyter Notebook + SSH-туннель на локальный порт + venv с torch
 - Режим `--dry-run` для просмотра плана
 - Красивый вывод с помощью `rich`
 
@@ -66,14 +67,14 @@ Dynamic: license-file
 pip install gpu-server-setup
 ```
 
-После установки команда `gpu-setup` будет доступна в терминале.
+После установки команда `gpu-server-setup` будет доступна в терминале.
 
 ---
 
 ## Использование
 
 ```bash
-gpu-setup [команда] [опции]
+gpu-server-setup [команда] [опции]
 ```
 
 ### Доступные команды
@@ -84,7 +85,10 @@ gpu-setup [команда] [опции]
 | `ssh`             | Только настройка SSH-доступа                  |
 | `ollama-install`  | Установка Ollama                              |
 | `model-pull`      | Загрузка модели                               |
-| `vscode`          | Открытие VS Code Remote                       |
+| `vscode`          | Открытие VS Code 
+colab                               |
+| `colab`          | upyter Notebook + локальный туннель 
+Remote                       |
 | `plan`            | Показать план действий                        |
 
 ---
@@ -132,7 +136,7 @@ gpu-setup [команда] [опции]
 
 **Пример:**
 ```bash
-gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
+gpu-server-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pem
 ```
 
 **Опции:**
@@ -152,37 +156,43 @@ gpu-setup model-pull llama3.2 --ip 123.45.67.89 --user ubuntu --pem-key ~/key.pe
 ### 1. Полная настройка со своим ключом (классический способ)
 
 ```bash
-gpu-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
+gpu-server-setup setup --ip 123.45.67.89 --user ubuntu --password "pass" --model llama3.2
 ```
 
 ### 2. Полная настройка с `.pem`-ключом (самый удобный для облака)
 
 ```bash
-gpu-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup setup --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ### 3. Только SSH с `.pem`-ключом
 
 ```bash
-gpu-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ssh --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 4. Установка Ollama с `.pem`-ключом
 
 ```bash
-gpu-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup ollama-install --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
 ### 5. Скачивание модели (после настройки SSH)
 
 ```bash
-gpu-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
+gpu-server-setup model-pull gemma2:27b --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem
 ```
 
-### 6. Просмотр плана
+### 6. Настройка colab
 
 ```bash
-gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
+gpu-server-setup colab --ip 123.45.67.89 --user ubuntu  --pem-key ~/key.pem (или --password) --port 9999
+```
+
+### 7. Просмотр плана
+
+```bash
+gpu-server-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-ubuntu.pem --model llama3.2
 ```
 
 ---
@@ -192,7 +202,7 @@ gpu-setup plan --ip 111.11.111.111 --user ubuntu --pem-key ~/Downloads/keyname-u
 Все команды поддерживают `--dry-run`:
 
 ```bash
-gpu-setup setup --ip IP --user user --pem-key key.pem --dry-run
+gpu-server-setup setup --ip IP --user user --pem-key key.pem --dry-run
 ```
 
 ---

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/gpu_setup/cli.py

@@ -3,9 +3,11 @@ from typing import Optional
 
 from .core import GPUServer, get_plan
 from .rich_console import header, plan_table, success, error
+import secrets
+
 
 app = typer.Typer(
-    name="gpu-setup",
+    name="gpu-server-setup",
     help="Один клик — удалённый GPU-сервер с Ollama + VS Code Remote",
     rich_markup_mode="rich",
     add_completion=True,
@@ -206,6 +208,36 @@ def vscode(
     )
     server.open_vscode()
 
+@app.command()
+def colab(
+    ip: str = typer.Option(..., "--ip"),
+    user: str = typer.Option(..., "--user"),
+    password: Optional[str] = typer.Option(None, "--password", hide_input=True),
+    port: int = typer.Option(8889, "--port"),
+    token: Optional[str] = typer.Option(None, "--token"),
+    dry_run: bool = typer.Option(False, "--dry-run"),
+    pem_key: Optional[str] = typer.Option(None, "--pem-key", "--key-file"),
+):
+    token = token or secrets.token_hex(16)
+    
+    server = GPUServer(
+        ip=ip,
+        user=user,
+        password=password,
+        external_key=pem_key,
+        dry_run=dry_run,
+    )
+    server.generate_key()
+    server.copy_pubkey()
+    server.setup_ssh_config()
+    server.test_connection()
+    
+    port = server.setup_colab(port=port, token=token)
+
+    success("Colab runtime готов!")
+    typer.echo(f"\nToken: [bold]{token}[/bold]")
+    typer.echo(f"Local URL: http://localhost:{port}/?token={token}")
+
 
 if __name__ == "__main__":
     app()

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/gpu_setup/core.py

@@ -310,7 +310,6 @@ class GPUServer:
             else:
                 success("Сервис Ollama успешно запущен (через systemctl)")
 
-        # Проверяем наличие модели
         _, stdout, _ = client.exec_command(f"ollama list | grep -q '{model}'")
         if stdout.channel.recv_exit_status() == 0:
             success(f"Модель {model} уже есть")
@@ -345,7 +344,111 @@ class GPUServer:
             success("VS Code Remote открыт")
         else:
             warning("Команда 'code' не найдена. Установи VS Code CLI")
+            
+    def setup_colab(self, port: int = 8889, token: str = "token123") -> int:
+        header("Настройка Colab (жёсткий режим)")
 
+        if self.dry_run:
+            console.print("[dim]DRY-RUN: colab setup[/dim]")
+            return port
+
+        def run_ssh(cmd: str, sudo: bool = False, print_output: bool = False) -> tuple[str, str]:
+            if sudo:
+                if self.password:
+                    remote_cmd = f"echo '{self.password}' | sudo -S {cmd}"
+                else:
+                    remote_cmd = f"sudo {cmd}"
+            else:
+                remote_cmd = cmd
+
+            if self.dry_run:
+                console.print(f"[dim]DRY-RUN: ssh {self.alias} '{remote_cmd}'[/dim]")
+                return "", ""
+
+            ssh_cmd = ["ssh", self.alias, remote_cmd]
+
+            try:
+                result = subprocess.run(
+                    ssh_cmd,
+                    capture_output=True,
+                    text=True,
+                )
+                out = result.stdout
+                err = result.stderr
+
+                if print_output and (out or err):
+                    if out:
+                        console.print(out, end="")
+                    if err:
+                        console.print(err, end="", style="red")
+
+                return out, err
+            except Exception as e:
+                error(f"SSH-ошибка: {e}")
+                return "", str(e)
+
+        header("Чистка окружения")
+        run_ssh("pkill -f jupyter || true", print_output=False)
+        run_ssh(f"fuser -k {port}/tcp || true", print_output=False)
+        subprocess.run(
+            ["pkill", "-f", f"{port}:localhost:{port}"],
+            stdout=subprocess.DEVNULL,
+            stderr=subprocess.DEVNULL,
+        )
+
+        header("Подготовка окружения")
+        run_ssh("apt update", sudo=True)
+        run_ssh("apt install -y python3-venv", sudo=True)
+
+        run_ssh("test -d ~/venv || python3 -m venv ~/venv")
+
+        pip_cmd = (
+            "bash -lc 'source ~/venv/bin/activate && "
+            "pip install --upgrade pip -q && "
+            "pip install -q jupyterlab notebook jupyter_server torch torchvision torchaudio'"
+        )
+        run_ssh(pip_cmd, print_output=False)   
+
+        header("Запуск Jupyter")
+        jupyter_cmd = (
+            f"setsid nohup ~/venv/bin/jupyter notebook "
+            f"--no-browser "
+            f"--ip=0.0.0.0 "
+            f"--port={port} "
+            f"--IdentityProvider.token={token} "
+            f"> /tmp/jupyter.log 2>&1 < /dev/null &"
+        )
+        run_ssh(jupyter_cmd)
+
+        header("Ожидание запуска Jupyter")
+
+        def wait_for_port() -> bool:
+            for _ in range(30):
+                out, _ = run_ssh(
+                    f"ss -tuln | grep -q '{port}' && echo yes || echo no"
+                )
+                if out.strip() == "yes":
+                    return True
+                time.sleep(1)
+            return False
+
+        if not wait_for_port():
+            error("Jupyter не запустился. Лог ниже:")
+            run_ssh("cat /tmp/jupyter.log || echo NO_LOG", print_output=True)
+            run_ssh("ps aux | grep -E 'jupyter|python' | grep -v grep", print_output=True)
+            sys.exit(1)
+
+        success("Jupyter запущен")
+
+        header("Создание SSH-туннеля")
+        tunnel_cmd = [
+            "ssh", "-N", "-L", f"{port}:localhost:{port}", self.alias
+        ]
+        subprocess.Popen(tunnel_cmd)
+
+        success("Туннель поднят")
+        return port
+        
 
 def get_plan(server: GPUServer, install_ollama: bool, model: Optional[str], no_vscode: bool) -> list:
     plan = [

{gpu_server_setup-0.4.8 → gpu_server_setup-0.4.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "gpu-server-setup"
-version = "0.4.8"
+version = "0.4.9"
 description = "Один клик — удалённый GPU-сервер с Ollama + VS Code Remote"
 readme = "README.md"
 license = {text = "MIT"}