[Chore] add arg to specify python version in dockerfile (#27)

Signed-off-by: Haiyang Shi <[email protected]>

Author: DwyaneShi

docker/Dockerfile_ubuntu2204

@@ -1,7 +1,19 @@
 FROM docker.io/library/ubuntu:22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
-RUN apt update && apt install -y git gcc make wget librdmacm-dev rdma-core libibverbs-dev libncurses5-dev libmount-dev libevent-dev libssl-dev python3-pybind11 python3-dev python3-pip libhiredis-dev liburing-dev
+RUN apt update && apt install -y git gcc make wget librdmacm-dev rdma-core libibverbs-dev libncurses5-dev libmount-dev libevent-dev libssl-dev libhiredis-dev liburing-dev pkg-config
+
+# Install Python
+ARG PYTHON_VERSION=3.12
+RUN apt install software-properties-common -y
+RUN add-apt-repository ppa:deadsnakes/ppa -y
+RUN apt update
+RUN apt install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv python3-pip
+RUN python${PYTHON_VERSION} -m ensurepip --upgrade
+RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 10
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 10
+RUN update-alternatives --config python3
+RUN python3 -m pip install --upgrade setuptools wheel pip pybind11
 
 # install cmake
 RUN cd /tmp && \