File tree 2 files changed +21
-0
lines changed
2 files changed +21
-0
lines changed Original file line number Diff line number Diff line change
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
+
Original file line number Diff line number Diff line change
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 . `
You can’t perform that action at this time.
0 commit comments