Spaces:
Runtime error
Runtime error
lucianotonet
commited on
Commit
•
135637e
1
Parent(s):
57f8000
Aprimorar suporte a GPU no Dockerfile
Browse filesSubstituiçã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.
- 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 |
|