fixed the dockerfile issue

Author: umechand-amd

.ci/docker/centos-rocm/Dockerfile

@@ -101,10 +101,13 @@ COPY ./common/install_cache.sh install_cache.sh
 ENV PATH /opt/cache/bin:$PATH
 RUN bash ./install_cache.sh && rm install_cache.sh
 
+WORKDIR /app
+RUN apt-get update && apt-get install -y git python3 python3-pip python3-setuptools
 RUN git clone https://github.com/ROCm/rocm-libraries.git
-RUN cd rocm-libraries/shared/origami/python
-RUN python setup.py build_ext --inplace
-RUN export PYTHONPATH=$(pwd):$PYTHONPATH
+RUN pip install nanobind
+WORKDIR /app/rocm-libraries/shared/origami/python
+RUN python3 setup.py build_ext --inplace
+ENV PYTHONPATH=/app/rocm-libraries/shared/origami/python:$PYTHONPATH
 
 # Include BUILD_ENVIRONMENT environment variable in image
 ARG BUILD_ENVIRONMENT

.ci/docker/ubuntu-rocm/Dockerfile

@@ -133,10 +133,14 @@ COPY ./common/install_openmpi.sh install_openmpi.sh
 RUN if [ -n "${CUDA_VERSION}" ]; then bash install_openmpi.sh; fi
 RUN rm install_openmpi.sh
 
+
+WORKDIR /app
+RUN apt-get update && apt-get install -y git python3 python3-pip python3-setuptools
 RUN git clone https://github.com/ROCm/rocm-libraries.git
-RUN cd rocm-libraries/shared/origami/python
-RUN python setup.py build_ext --inplace
-RUN export PYTHONPATH=$(pwd):$PYTHONPATH
+RUN pip install nanobind
+WORKDIR /app/rocm-libraries/shared/origami/python
+RUN python3 setup.py build_ext --inplace
+ENV PYTHONPATH=/app/rocm-libraries/shared/origami/python:$PYTHONPATH
 
 # Include BUILD_ENVIRONMENT environment variable in image
 ARG BUILD_ENVIRONMENT