Fix MPI

Dockerfile_cuda

@@ -50,8 +50,9 @@ RUN mamba install -c nvidia -c conda-forge -c bioconda --yes \
         'pennylane-lightning-gpu' \
         'mpi4py' \
         'openmpi' \
+        'nccl' \
         "cuda-version=${CONDA_OVERRIDE_CUDA%%.*}" \
-    && mamba env config vars set "CUQUANTUM_ROOT=${CONDA_PREFIX}" \
+    && mamba env config vars set "CUQUANTUM_ROOT=${CONDA_DIR}" "OMPI_MCA_opal_cuda_support=true" "UCX_MEMTYPE_CACHE=n" \
     && echo "libnvidia-opencl.so.1" > "${CONDA_DIR}/etc/OpenCL/vendors/nvidia.icd" \
     && mamba clean --all -f -y \
     && pip install --no-cache-dir --extra-index-url="https://pypi.nvidia.com" \