add triton solution

This commit is contained in:
Yuekai Zhang
2025-07-22 06:50:13 -07:00
parent b048a2d6db
commit 5427c274e3
18 changed files with 3448 additions and 0 deletions

View File

@@ -0,0 +1,6 @@
FROM nvcr.io/nvidia/tritonserver:25.06-trtllm-python-py3
RUN apt-get update && apt-get install -y cmake
RUN git clone https://github.com/pytorch/audio.git && cd audio && git checkout c670ad8 && PATH=/usr/local/cuda/bin:$PATH python3 setup.py develop
COPY ./requirements.txt /workspace/requirements.txt
RUN pip install -r /workspace/requirements.txt
WORKDIR /workspace