R
RunPod•5w ago
Dronte

How to build on top of runpod dockerfile?

Hi. My dockerfile is +- this: 
FROM runpod/pytorch:2.8.0-py3.11-cuda12.8.1-cudnn-devel-ubuntu22.04
RUN pip install ninja
ENV MAX_JOBS=4
RUN MAX_JOBS=4 bash -c "ulimit -v 10485760 && pip install --no-build-isolation flash-attn"
FROM runpod/pytorch:2.8.0-py3.11-cuda12.8.1-cudnn-devel-ubuntu22.04
RUN pip install ninja
ENV MAX_JOBS=4
RUN MAX_JOBS=4 bash -c "ulimit -v 10485760 && pip install --no-build-isolation flash-attn"
but I could not find out how to make it runnable with pods, any time I try, my container fails to start. What should i write to be able to ssh into the pod?
1 Reply
riverfog7
riverfog7•4w ago
build that docker file docker build and then tag it and push it to the docker hub and then use that image for runpod

Did you find this page helpful?