From 1360c42fe63761944c78bc5828f538ef7eff58e7 Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Thu, 2 Jul 2026 10:38:50 +0800 Subject: [PATCH] [UX] Include NVTX in cuda.txt (#47319) Signed-off-by: Jee Jee Li --- requirements/cuda.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements/cuda.txt b/requirements/cuda.txt index acfbe7d048..5545d3344f 100644 --- a/requirements/cuda.txt +++ b/requirements/cuda.txt @@ -15,7 +15,8 @@ flashinfer-cubin==0.6.13 apache-tvm-ffi==0.1.9 tilelang==0.1.9 nvidia-cudnn-frontend>=1.19.1 - +# Required for LLM_NVTX_SCOPES_FOR_PROFILING=1 +nvtx==0.2.15 # Required for faster safetensors model loading fastsafetensors >= 0.3.2