mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-23 19:30:12 +01:00
Rework Ollama base
This commit is contained in:
parent
37c08d4b55
commit
60d862ec0d
1 changed files with 9 additions and 2 deletions
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue