mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-25 12:20:12 +01:00
39 lines
1.2 KiB
Docker
39 lines
1.2 KiB
Docker
# Base Image
|
|
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://github.com/ollama/ollama/releases/download/v0.2.8/ollama-linux-amd64 -o /usr/bin/ollama && \
|
|
chmod +x /usr/bin/ollama
|
|
|
|
COPY settings.json /root/.local/share/code-server/User/settings.json
|
|
COPY tasks.json /root/.local/share/code-server/User/tasks.json
|
|
|
|
# Start Scripts
|
|
COPY --from=scripts start.sh /
|
|
RUN chmod +x /start.sh
|
|
|
|
# Set environment variables for runtime
|
|
ENV OLLAMA_HOST=0.0.0.0
|
|
ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
|
|
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64
|
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
|
|
# Expose the required port
|
|
EXPOSE 11434
|
|
|
|
|
|
CMD ["/start.sh"]
|