Trouble installing lightning-gpu with mpi support under Python3.10

PennyLane Help
2

Hi @justin6626,

I’m not sure, I’m checking with the team so I’ll let you know what I hear from them.

Alternatively you can create a Docker image with the following steps:

FROM nvcr.io/nvidia/cuquantum-appliance:24.08-x86_64 as base

ENV DEBIAN_FRONTEND=noninteractive
ENV PATH=/usr/local/cuda/bin:$PATH
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

RUN apt-get update && apt-get install -y build-essential openmpi-bin libopenmpi-dev

RUN git clone https://github.com/PennyLaneAI/pennylane-lightning.git && \
cd pennylane-lightning && \
pip install -r requirements.txt --upgrade && \
python -m pip install . && \
PL_BACKEND="lightning_gpu" python scripts/configure_pyproject_toml.py && \
CMAKE_ARGS="-DENABLE_MPI=ON" python -m pip install . -vv

Note that at the moment this doesn’t work with editable mode installation of Lightning_GPU-MPI.

Let me know if this works for you!

1 Like