Adiciona suporte ao NVIDIA Toolkit no Dockerfile

Inclui a instalação do NVIDIA Container Toolkit para habilitar o suporte a GPUs. Ajustes na instalação de dependências e configuração do runtime visam otimizar o ambiente para aplicações que requerem aceleração gráfica. A remoção do comando que puxa o modelo no início evita possíveis conflitos de execução enquanto prepara o contêiner, aprimorando a robustez do processo.

Dockerfile

@@ -1,7 +1,10 @@
 FROM ollama/ollama:latest
 
 # Instalar dependências necessárias
-RUN apt-get update && apt-get install curl -y
+RUN apt-get update && apt-get install -y curl
+RUN apt-get install -y 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
@@ -11,7 +14,7 @@ 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
+# RUN ollama serve & sleep 5 && ollama pull lucianotonet/llamaclaude && pkill ollama
 
 # Expor a porta do Ollama
 EXPOSE 11434