15 lines
459 B
Docker
15 lines
459 B
Docker
FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
|
|
|
|
RUN apt-get update && apt-get install -y python3 python3-pip procps psmisc tini && \
|
|
pip3 install --no-cache-dir torch && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
COPY parent.py gpu_worker_fixed.py ./
|
|
COPY observe.sh /usr/local/bin/observe.sh
|
|
COPY prestop.sh /usr/local/bin/prestop.sh
|
|
RUN chmod +x /usr/local/bin/observe.sh /usr/local/bin/prestop.sh
|
|
|
|
ENTRYPOINT ["tini", "--"]
|
|
CMD ["sleep", "infinity"]
|