mirror of
https://github.com/ghndrx/kubeflow-pipelines.git
synced 2026-02-10 06:45:13 +00:00
22 lines
374 B
Docker
22 lines
374 B
Docker
FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
|
|
|
|
WORKDIR /app
|
|
|
|
# Install dependencies
|
|
RUN pip install --no-cache-dir \
|
|
runpod \
|
|
transformers \
|
|
datasets \
|
|
accelerate \
|
|
boto3 \
|
|
scikit-learn \
|
|
scipy
|
|
|
|
# Copy handler
|
|
COPY handler.py /app/handler.py
|
|
|
|
# Set environment variables
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
CMD ["python", "-u", "handler.py"]
|