gpu-server-setup 0.4.0

Один клик — удалённый GPU-сервер с Ollama + VS Code Remote

# Gpu-setup — один клик до удалённого GPU-сервера с Ollama и VS Code Remote

Автоматическая настройка SSH, установка [Ollama](https://ollama.com/), загрузка LLM-моделей и открытие VS Code Remote на удалённом GPU-сервере — всё одной командой.

---

## Возможности

- **Генерация SSH-ключей** (ed25519 / RSA) и автоматическое копирование публичного ключа на сервер.
- **Настройка `~/.ssh/config`** с алиасом для удобного подключения.
- **Установка Ollama** (через официальный скрипт) с запуском сервиса.
- **Загрузка моделей** (например, `llama3.2`, `gemma2`) с живым выводом прогресса.
- **Автоматический запуск VS Code Remote** (если установлен `code` CLI).
- **Режим `--dry-run`** для безопасного просмотра плана действий.
- **Красивый вывод** с помощью `rich`.

---

## Предварительные требования

- Python 3.8+ (рекомендуется 3.10)
- Удалённый сервер с Linux (Ubuntu/Debian/CentOS) и предустановленным Python
- Доступ по SSH с паролем (для первого копирования ключа)
- На локальной машине:
  - `ssh-keygen` (обычно уже есть)
  - `ssh` клиент
  - (опционально) `code` для VS Code Remote (если планируете открывать редактор)

---

## Установка

```bash
# Клонируем репозиторий
git clone https://github.com/your-username/gpu-setup.git
cd gpu-setup

# Устанавливаем в виртуальное окружение (рекомендуется)
python -m venv venv
source venv/bin/activate   # или venv\Scripts\activate на Windows

# Устанавливаем зависимости
pip install -r requirements.txt

# Устанавливаем пакет в режиме разработки (опционально)
pip install -e .

После этого команда gpu-setup будет доступна в терминале.

---

Использование

Общий синтаксис:

gpu-setup [команда] [опции]

Доступные команды

Команда	Описание
setup	Полная настройка: SSH, Ollama, модель, VS Code
ssh	Только настройка SSH-ключей и алиаса
ollama-install	Установка Ollama на сервер
model-pull	Загрузка указанной модели (после установки Ollama)
vscode	Открытие VS Code Remote (если code установлен)
plan	Показать план действий без выполнения (dry-run)

---

Подробное описание команд

gpu-setup setup

Выполняет все шаги: генерация ключа, копирование, настройка SSH-конфига, установка Ollama, загрузка модели, открытие VS Code.

Опции:

• --ip (обязательный) — IP сервера
• --user (обязательный) — имя пользователя
• --password (обязательный для первого подключения) — пароль (можно ввести интерактивно)
• --model — модель для скачивания (например llama3.2)
• --install-ollama / --no-install — устанавливать Ollama (по умолчанию --install-ollama)
• --no-vscode — не открывать VS Code (по умолчанию открывает)
• --dry-run — только показать план
• --key-type — тип ключа: ed25519 (по умолчанию) или rsa

---

gpu-setup ssh

Только настройка SSH-доступа по ключу.

Опции:

• --ip (обязательный)
• --user (обязательный)
• --password (обязательный для первого копирования)
• --dry-run
• --key-type

---

gpu-setup ollama-install

Устанавливает Ollama на сервер (требуется пароль для sudo).

Опции:

• --ip (обязательный)
• --user (обязательный)
• --password (обязательный, т.к. нужен sudo)
• --dry-run

---

gpu-setup model-pull MODEL

Скачивает указанную модель через Ollama. Если сервис Ollama не запущен, пытается запустить его автоматически.

Аргументы:

• MODEL — имя модели (например llama3.2)

Опции:

• --ip (обязательный)
• --user (обязательный)
• --password (необязателен, если SSH уже настроен; нужен только для запуска сервиса)
• --dry-run

---

gpu-setup vscode

Открывает текущую сессию в VS Code Remote.

Опции:

• --ip (обязательный)
• --user (обязательный)
• --password (не требуется, если ключ настроен)
• --dry-run

---

gpu-setup plan

Показывает, какие шаги будут выполнены при вызове setup с указанными параметрами.

Опции:

• --ip (обязательный)
• --user (обязательный)
• --model
• --install-ollama / --no-install
• --no-vscode

---

Примеры использования

1. Полная настройка (самый частый вариант)

gpu-setup setup --ip 195.208.16.2 --user user --password sWDoAVym --model llama3.2

2. Только SSH-ключи + config

gpu-setup ssh --ip 195.208.16.2 --user user --password sWDoAVym

3. Только установить Ollama

gpu-setup ollama-install --ip 195.208.16.2 --user user --password sWDoAVym

4. Только скачать модель (пароль уже не нужен, если SSH настроен)

gpu-setup model-pull llama3.2 --ip 195.208.16.2 --user user

5. Только открыть VS Code Remote

gpu-setup vscode --ip 195.208.16.2 --user user

6. Посмотреть план перед запуском

gpu-setup plan --ip 195.208.16.2 --user user --model llama3.2

---

Dry-run

Любая команда поддерживает флаг --dry-run.
В этом режиме ничего не изменяется, а только выводится, что было бы сделано.

gpu-setup setup --ip 195.208.16.2 --user user --dry-run

---

Устранение неполадок

• ssh: Could not resolve hostname — проверьте IP и доступность сервера.
• Ошибка при копировании ключа — убедитесь, что пароль верный и на сервере разрешена аутентификация по паролю (PasswordAuthentication yes в /etc/ssh/sshd_config).
• Ollama не устанавливается — возможно, требуется вручную разрешить sudo без пароля или передать пароль в опции.
• VS Code не открывается — установите CLI-команду code (в VS Code: Shift+Ctrl+P → Install 'code' command in PATH).

---

Лицензия

MIT License. См. файл LICENSE.

---

Вклад

PR и issues приветствуются!