lucianotonet commited on
Commit
135637e
1 Parent(s): 57f8000

Aprimorar suporte a GPU no Dockerfile

Browse files

Substituição do pacote de instalação para melhor compatibilidade com a gestão de containers NVIDIA. A adição do `nvidia-docker2` garante uma integração mais robusta com o Docker, enquanto o uso do `--no-install-recommends` para o `nvidia-container-toolkit` minimiza a instalação de pacotes desnecessários, tornando a imagem mais leve e eficiente. Essas mudanças visam otimizar o uso de recursos ao trabalhar com GPUs em ambientes de contêiner.

Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM ollama/ollama:latest
2
 
3
  # Instalar dependências necessárias
4
  RUN apt-get update && apt-get install -y curl
5
- RUN apt-get install -y nvidia-container-toolkit
6
 
7
  RUN nvidia-ctk runtime configure --runtime=docker
8
 
 
2
 
3
  # Instalar dependências necessárias
4
  RUN apt-get update && apt-get install -y curl
5
+ RUN apt-get install -y nvidia-docker2 && apt-get install -y --no-install-recommends nvidia-container-toolkit
6
 
7
  RUN nvidia-ctk runtime configure --runtime=docker
8