Skip to content

Commit 51612ec

Browse files
Murat MagomedovMurat Magomedov
Murat Magomedov
authored and
Murat Magomedov
committed
added TRRT_LLM dockerfile reference for pods
1 parent 66d5f97 commit 51612ec

File tree

2 files changed

+21
-0
lines changed

2 files changed

+21
-0
lines changed

Diff for: pod-image/Dockerfile

+18
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
# Start with Runpod Python 3.10 base image
2+
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04
3+
4+
# Avoid prompts from apt
5+
ENV DEBIAN_FRONTEND=noninteractive
6+
7+
# Install system dependencies and Python
8+
RUN apt-get update -y && \
9+
apt-get install -y git libopenmpi-dev && \
10+
apt-get clean && \
11+
rm -rf /var/lib/apt/lists/*
12+
13+
# Update Torch
14+
RUN pip install torch==2.4.0 torchvision==0.19.0
15+
16+
# Install TensorRT-LLM
17+
RUN pip install tensorrt_llm==0.13.0 -U --extra-index-url https://pypi.nvidia.com
18+

Diff for: pod-image/README.md

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

0 commit comments

Comments
 (0)