Skip to content

Commit

Permalink
HPCC-32142 Add SVM to platform-ml and platform-gnn Docker images
Browse files Browse the repository at this point in the history
Add missing install instruction for SVM to the relevant Dockerfiles

Signed-off-by: Attila Vamos <[email protected]>
  • Loading branch information
AttilaVamos committed Jun 27, 2024
1 parent ca1439c commit 3fc8be8
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 0 deletions.
2 changes: 2 additions & 0 deletions dockerfiles/platform-gnn-gpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -85,6 +85,8 @@ RUN apt clean && \
apt autoremove && \
apt-get update

RUN apt-get install -y libsvm-dev libsvm-tools

RUN apt-get install -y python3-pip --fix-missing
RUN python3 -m pip --no-cache-dir install \
setuptools
Expand Down
1 change: 1 addition & 0 deletions dockerfiles/platform-gnn/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ FROM ${DOCKER_REPO}/platform-core:${BUILD_LABEL}
USER root

RUN apt-get update -y && apt-get install -y python3-pip --fix-missing
RUN apt-get install -y libsvm-dev libsvm-tools
RUN python3 -m pip install --upgrade pip

RUN pip3 install \
Expand Down
1 change: 1 addition & 0 deletions dockerfiles/platform-ml/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ USER root

# Machine Learning Dependencies
RUN apt-get update && apt-get install -y python3 python3-pip --fix-missing
RUN apt-get install -y libsvm-dev libsvm-tools
RUN python3 -m pip install --upgrade pip
RUN pip3 install \
scikit-learn
Expand Down

0 comments on commit 3fc8be8

Please sign in to comment.