Rework Ollama base

This commit is contained in:
Madiator2011 2024-07-13 20:21:22 +02:00 committed by GitHub
parent 37c08d4b55
commit 60d862ec0d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,14 +1,21 @@
# Base Image
FROM madiator2011/better-pytorch:cuda12.1
FROM madiator2011/better-pytorch:cuda12.4
# Install necessary dependencies
RUN apt-get update && \
apt-get install -y --no-install-recommends \
curl \
lshw \
&& rm -rf /var/lib/apt/lists/*
RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb && \
dpkg -i cuda-keyring_1.1-1_all.deb && \
apt-get update && \
apt-get -y install cuda-toolkit-12-4 && \
rm cuda-keyring_1.1-1_all.deb
# Download and install Ollama
RUN curl -L https://ollama.com/download/ollama-linux-amd64 -o /usr/bin/ollama && \
RUN curl -L https://github.com/ollama/ollama/releases/download/v0.2.2/ollama-linux-amd64 -o /usr/bin/ollama && \
chmod +x /usr/bin/ollama
COPY settings.json /root/.local/share/code-server/User/settings.json