Skip to content

Commit

Permalink
added TRRT_LLM dockerfile reference for pods
Browse files Browse the repository at this point in the history
  • Loading branch information
Murat Magomedov authored and Murat Magomedov committed Oct 10, 2024
1 parent 66d5f97 commit 51612ec
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 0 deletions.
18 changes: 18 additions & 0 deletions pod-image/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
# Start with Runpod Python 3.10 base image
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04

# Avoid prompts from apt
ENV DEBIAN_FRONTEND=noninteractive

# Install system dependencies and Python
RUN apt-get update -y && \
apt-get install -y git libopenmpi-dev && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*

# Update Torch
RUN pip install torch==2.4.0 torchvision==0.19.0

# Install TensorRT-LLM
RUN pip install tensorrt_llm==0.13.0 -U --extra-index-url https://pypi.nvidia.com

3 changes: 3 additions & 0 deletions pod-image/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
Ready to use TensorRT-LLM image for a Pod, just pull image from: *muratmagomedov/runpod-trrt:v0.13.0*
Test: `python3 -c "import tensorrt_llm"`
To build yourself: `docker build -t <username/image> --platform linux/amd64 .`

0 comments on commit 51612ec

Please sign in to comment.