pip install -e . does not work

[wangkuiyi]

System Info

x86, H100, Ubuntu

Who can help?

No response

Information

• The official example scripts
• My own modified scripts

Tasks

• An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
• My own task or dataset (give details below)

Reproduction

On a system running CUDA 12.3 and H100, I installed the dependencies by running scripts referred to by Dockerfile.multi:

TensorRT-LLM/docker/Dockerfile.multi

Lines 8 to 51 in 0ab9d17

	# https://www.gnu.org/software/bash/manual/html_node/Bash-Startup-Files.html
	# The default values come from `nvcr.io/nvidia/pytorch`
	ENV BASH_ENV=${BASH_ENV:-/etc/bash.bashrc}
	ENV ENV=${ENV:-/etc/shinit_v2}
	SHELL ["/bin/bash", "-c"]
	FROM base as devel
	COPY docker/common/install_base.sh install_base.sh
	RUN bash ./install_base.sh && rm install_base.sh
	COPY docker/common/install_cmake.sh install_cmake.sh
	RUN bash ./install_cmake.sh && rm install_cmake.sh
	COPY docker/common/install_ccache.sh install_ccache.sh
	RUN bash ./install_ccache.sh && rm install_ccache.sh
	# Download & install internal TRT release
	ARG TRT_VER
	ARG CUDA_VER
	ARG CUDNN_VER
	ARG NCCL_VER
	ARG CUBLAS_VER
	COPY docker/common/install_tensorrt.sh install_tensorrt.sh
	RUN bash ./install_tensorrt.sh \
	--TRT_VER=${TRT_VER} \
	--CUDA_VER=${CUDA_VER} \
	--CUDNN_VER=${CUDNN_VER} \
	--NCCL_VER=${NCCL_VER} \
	--CUBLAS_VER=${CUBLAS_VER} && \
	rm install_tensorrt.sh
	# Install latest Polygraphy
	COPY docker/common/install_polygraphy.sh install_polygraphy.sh
	RUN bash ./install_polygraphy.sh && rm install_polygraphy.sh
	# Install mpi4py
	COPY docker/common/install_mpi4py.sh install_mpi4py.sh
	RUN bash ./install_mpi4py.sh && rm install_mpi4py.sh
	# Install PyTorch
	ARG TORCH_INSTALL_TYPE="skip"
	COPY docker/common/install_pytorch.sh install_pytorch.sh
	RUN bash ./install_pytorch.sh $TORCH_INSTALL_TYPE && rm install_pytorch.sh

by setting ENV to ~/.bashrc.

This allowed me to run the following command to build TensorRT-LLM from source code:

pip install -e . --extra-index-url https://pypi.nvidia.com

The building process is very fast, which does not look right, because it usually takes 40 minutes for build_wheel.py to build everything.

After the building, pip list shows that tensorrt-llm is installed.

$ pip list | grep tensorrt
tensorrt                 9.2.0.post12.dev5
tensorrt-llm             0.9.0.dev2024020600 /root/TensorRT-LLM

However, importing it would error:

$ pythonon3 -c 'import tensorrt_llm'
Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "/root/TensorRT-LLM/tensorrt_llm/__init__.py", line 44, in <module>
    from .hlapi.llm import LLM, ModelConfig
  File "/root/TensorRT-LLM/tensorrt_llm/hlapi/__init__.py", line 1, in <module>
    from .llm import LLM, ModelConfig
  File "/root/TensorRT-LLM/tensorrt_llm/hlapi/llm.py", line 17, in <module>
    from ..executor import (GenerationExecutor, GenerationResult,
  File "/root/TensorRT-LLM/tensorrt_llm/executor.py", line 11, in <module>
    import tensorrt_llm.bindings as tllm
ModuleNotFoundError: No module named 'tensorrt_llm.bindings'

Expected behavior

My project requires me to build the main branch of TensorRT-LLM. It would be great if pip install could work, so I could declare TensorRT-LLM as a dependency in my project's pyproject.toml file.

actual behavior

I had to build TensorRT-LLM by invoking build_wheel.py as in https://github.com/NVIDIA/TensorRT-LLM/blob/main/docs/source/build_from_source.md#build-tensorrt-llm

additional notes

I was able to build vLLM with the CUDA kernels using pip -e .. Not sure if we could take their build setup as a reference.

[TobyGE]

to install the main brach, you can use the following command
pip3 install tensorrt_llm -U --pre --extra-index-url https://pypi.nvidia.com

also check the readme for more detail

[wangkuiyi]

That doesn’t work. As described above, the project requires TensorRT-LLM built from the main branch.

[TobyGE]

I used and it worked, my current version is dev2024020600 your cuda version is too high, follow the instruction and use 12.1

…

On Thu, Feb 8, 2024 at 00:00 Yi Wang ***@***.***> wrote: That doesn’t work. As described above, the project requires TensorRT-LLM built from the main branch. — Reply to this email directly, view it on GitHub <#1065 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AFQTIVABGN6HGT6GQGGLPSDYSSA2JAVCNFSM6AAAAABC6QXWTSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTSMZTGUZTSNRQGM> . You are receiving this because you commented.Message ID: ***@***.***>

[wangkuiyi]

It doesn’t build the main branch. It installs a most recent pre release version. On Thu, Feb 8, 2024 at 1:17 AM Yingqiang Ge ***@***.***> wrote:

…

I used and worked On Thu, Feb 8, 2024 at 00:00 Yi Wang ***@***.***> wrote: > That doesn’t work. As described above, the project requires TensorRT-LLM > built from the main branch. > > — > Reply to this email directly, view it on GitHub > < #1065 (comment)>, > or unsubscribe > < https://github.com/notifications/unsubscribe-auth/AFQTIVABGN6HGT6GQGGLPSDYSSA2JAVCNFSM6AAAAABC6QXWTSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTSMZTGUZTSNRQGM> > . > You are receiving this because you commented.Message ID: > ***@***.***> > — Reply to this email directly, view it on GitHub <#1065 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AAL2DZ4BWZNGI6DS5MK2Y43YSSJ2HAVCNFSM6AAAAABC6QXWTSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTSMZTGY2TEMRSGE> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[wangkuiyi]

I looks like pip install -e . does not automatically trigger the buiding of the Python binding of the C++ runtime.

[jdemouth-nvidia]

@Shixiaowei02 , can you help with that issue, please?

[jdemouth-nvidia]

@wangkuiyi , for your information, @Shixiaowei02 is based in China. It means that he won't be able to work on this issue before the end of the break for the Chinese New Year.

[wangkuiyi]

Thank you @jdemouth-nvidia and @Shixiaowei02 ! No rush please. It is totally fine after the lunar new year.

[Shixiaowei02]

I am working on fixing this issue now. Thanks for your support!

[ekagra-ranjan]

Thanks! I am also facing this issue.

[andyluo7]

I am facing the same issue.

[Shixiaowei02]

Can you use these two commands to temporarily bypass this issue? We will fix this issue in the near future and synchronize it to the main branch. Thank you! @wangkuiyi

python3 scripts/build_wheel.py --trt_root /usr/local/tensorrt
pip3 install -e .

[lifelongeeek]

After build wheel & editable install, I still got the same error

    import tensorrt_llm.bindings as tllm
ModuleNotFoundError: No module named 'tensorrt_llm.bindings'