diff --git a/torchserve/Dockerfile b/torchserve/Dockerfile index 001217ca..22dc5b44 100644 --- a/torchserve/Dockerfile +++ b/torchserve/Dockerfile @@ -8,7 +8,6 @@ RUN apt-get update && \ DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \ ca-certificates \ curl \ - python3-pip \ vim \ sudo \ default-jre \ @@ -17,6 +16,9 @@ RUN apt-get update && \ build-essential \ && rm -rf /var/lib/apt/lists/* +RUN conda install python=3.8.13 -y + + # install python dependencies RUN pip install openmim RUN pip install torch==2.0.0