Lightweight docker image for inference generation.

Hello All, I am currently using

pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime

pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime

image for my servreless endpoint. The issue is that my Github action to build and push the docker image fails due

ERROR: Could not install packages due to an OSError: [Errno 28] No space left on device

ERROR: Could not install packages due to an OSError: [Errno 28] No space left on device

Is there any recommended lightweight docker image that I can use?

Runpod•2y ago•

6 replies

Lightweight docker image for inference generation.

Hello All, I am currently using

pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime

pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime

image for my servreless endpoint. The issue is that my Github action to build and push the docker image fails due

ERROR: Could not install packages due to an OSError: [Errno 28] No space left on device

ERROR: Could not install packages due to an OSError: [Errno 28] No space left on device

Is there any recommended lightweight docker image that I can use?

Join the Discord to ask follow-up questions and connect with the community

We're a community of enthusiasts, engineers, and enterprises, all sharing insights on AI, Machine Learning and GPUs!

21,906 Members