mirror of
https://github.com/ghndrx/kubeflow-pipelines.git
synced 2026-02-10 06:45:13 +00:00
Use Tailscale endpoints, add RunPod Docker build files
This commit is contained in:
@@ -2,20 +2,17 @@ FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# Copy requirements first for better caching
|
||||
COPY requirements.txt /app/requirements.txt
|
||||
|
||||
# Install dependencies
|
||||
RUN pip install --no-cache-dir \
|
||||
runpod \
|
||||
transformers \
|
||||
datasets \
|
||||
accelerate \
|
||||
boto3 \
|
||||
scikit-learn \
|
||||
scipy
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy handler
|
||||
COPY handler.py /app/handler.py
|
||||
|
||||
# Set environment variables
|
||||
ENV PYTHONUNBUFFERED=1
|
||||
ENV HF_HOME=/tmp/huggingface
|
||||
|
||||
CMD ["python", "-u", "handler.py"]
|
||||
|
||||
Reference in New Issue
Block a user