Skip to content

Commit 789e471

Browse files
committed
fix: install runpod package in task runtime images
1 parent 07e7fee commit 789e471

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

‎runpod/runtimes/task/Dockerfile‎

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,14 @@ ENV DEBIAN_FRONTEND=noninteractive \
77
TZ=Etc/UTC \
88
PYTHONUNBUFFERED=1
99

10-
RUN pip install --no-cache-dir cloudpickle
10+
RUN apt-get update && apt-get install -y --no-install-recommends \
11+
git ca-certificates \
12+
&& apt-get clean && rm -rf /var/lib/apt/lists/*
1113

12-
COPY runpod/runtimes/task/runner.py /runner.py
14+
# install the runpod package: task payloads carry the user's full module
15+
# source, which imports runpod (decorators, nested .remote calls)
16+
COPY . /src
17+
RUN pip install --no-cache-dir /src cloudpickle && rm -rf /src
1318

1419
EXPOSE 8080
15-
CMD ["python", "/runner.py"]
20+
CMD ["python", "-m", "runpod.runtimes.task.runner"]

0 commit comments

Comments
 (0)