FROM ollama/ollama:latest

# Instalar dependências necessárias
RUN apt-get update && apt-get install -y curl
RUN apt-get install -y nvidia-docker2 && apt-get install -y --no-install-recommends nvidia-container-toolkit

RUN nvidia-ctk runtime configure --runtime=docker

# https://huggingface.co/docs/hub/spaces-sdks-docker-first-demo
RUN useradd -m -u 1000 user

USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH \
    OLLAMA_HOST=0.0.0.0

# RUN ollama serve & sleep 5 && ollama pull lucianotonet/llamaclaude && pkill ollama

# Expor a porta do Ollama
EXPOSE 11434