Запуск Docker-образа

1. Подготовьте образ

Используйте базовый образ с поддержкой CUDA:

FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04

RUN apt-get update && apt-get install -y python3-pip
RUN pip3 install torch torchvision

COPY . /app
WORKDIR /app
CMD ["python3", "train.py"]

2. Загрузите образ в реестр

Опубликуйте образ в Docker Hub или приватном реестре:

docker build -t youruser/myapp:latest .
docker push youruser/myapp:latest

3. Укажите образ при создании инстанса

В панели управления при создании инстанса выберите «Свой Docker-образ» и введите имя образа. Инстанс запустит контейнер с доступом к GPU автоматически.

4. Проверьте работу GPU в контейнере

import torch
print(torch.cuda.is_available())  # True
print(torch.cuda.get_device_name(0))