generated from runpod-workers/worker-template
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
added TRRT_LLM dockerfile reference for pods
- Loading branch information
Murat Magomedov
authored and
Murat Magomedov
committed
Oct 10, 2024
1 parent
66d5f97
commit 51612ec
Showing
2 changed files
with
21 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 .` |