Як налаштувати Docker для запуску моделей AI локально

Введення

Docker — це інструмент для віртуалізації застосунків, який дозволяє запускати моделі AI в ізольованих середовищах. Завдяки цьому ви можете легко керувати залежностями та середовищами, уникаючи конфліктів між різними проектами. У цій статті ми розглянемо, як налаштувати Docker для запуску моделей AI локально.

Вимоги попередні

Перед початком налаштування Docker вам знадобиться:

Встановлений Docker Desktop на вашому комп'ютері.
Основні знання про Docker та контейнеризацію.
Модель AI, яку ви хочете запустити.

Встановлення Docker Desktop

Якщо у вас ще не встановлено Docker Desktop, ви можете це зробити, дотримуючись інструкцій на сайті Docker.

Створення файлу Dockerfile

Аби запустити модель AI в Docker, вам потрібно створити файл Dockerfile, який визначає середовище та залежності, необхідні для запуску моделі. Нижче наведено приклад Dockerfile для моделі AI на основі Python:

# Використовуйте офіційний образ Python
FROM python:3.9-slim

# Встановіть змінну середовища LANG
ENV LANG C.UTF-8

# Оновіть пакети та встановіть залежності
RUN apt-get update && apt-get install -y \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

# Створіть робочу папку
WORKDIR /app

# Скопіюйте вимоги до робочої папки
COPY requirements.txt .

# Встановіть залежності Python
RUN pip install --no-cache-dir -r requirements.txt

# Скопіюйте решту файлів до робочої папки
COPY . .

# Вкажіть порт, на якому застосунок буде слухати
EXPOSE 8000

# Вкажіть команду для запуску застосунку
CMD ["python", "app.py"]

Створення файлу requirements.txt

Файл requirements.txt містить список залежностей Python, необхідних для запуску моделі AI. Приклад файлу requirements.txt може виглядати наступним чином:

numpy==1.21.2
pandas==1.3.3
tensorflow==2.6.0
flask==2.0.1

Збирання образу Docker

Аби зібрати образ Docker, використайте наступну команду в терміналі:

docker build -t ai-model .

Ця команда зібере образ Docker на основі Dockerfile та призначить йому тег ai-model.

Запуск контейнера Docker

Після зборки образу ви можете запустити контейнер Docker за допомогою наступної команди:

docker run -p 8000:8000 ai-model

Ця команда запустить контейнер Docker та відобразить порт 8000 контейнера на порт 8000 хоста.

Тестування моделі AI

Аби перевірити, чи модель AI працює правильно, ви можете використати інструмент curl або відкрити браузер і перейти за адресою http://localhost:8000.

Керування контейнерами Docker

Docker надає кілька команд для керування контейнерами. Нижче наведено деякі з них:

Список запущених контейнерів:
```
docker ps
```
Список всіх контейнерів (включно з зупиненими):
```
docker ps -a
```
Зупинка контейнера:
```
docker stop <container_id>
```
Видалення контейнера:
```
docker rm <container_id>
```

Підсумок

Docker — це потужний інструмент для запуску моделей AI в ізольованих середовищах. Завдяки йому ви можете легко керувати залежностями та середовищами, уникаючи конфліктів між різними проектами. У цій статті ми розглянули, як налаштувати Docker для запуску моделей AI локально. Сподіваємося, що ці інформація буде для вас корисною!