Skip to content

Commit

Permalink
Removed torch and Horovod in Dockerfile
Browse files Browse the repository at this point in the history
Since ludwig[full] will install torch and Ray. Ludwig has strong support for Ray, a framework for distributed computing, so we are not going to use Horovod. Ray doesn't support python 3.11 now.
  • Loading branch information
qchiujunhao committed Mar 26, 2024
1 parent 8cc5c50 commit 70adf90
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,15 +1,14 @@
FROM python:3.10-slim

ARG VERSION=0.6.1
ARG VERSION=0.10

RUN apt-get -y update && apt-get install -y --no-install-recommends build-essential cmake git unzip

RUN export HOROVOD_WITH_PYTORCH=1 && \
pip install -U pip && \
pip install 'torch' 'git+https://github.com/goeckslab/model-unpickler.git' && \
RUN pip install -U pip && \
pip install 'git+https://github.com/goeckslab/model-unpickler.git' && \
pip install 'git+https://github.com/goeckslab/smart-report.git@17df590f3ceb065add099f37b4874c85bd275014' && \
pip install 'horovod[pytorch]' && \
pip install 'ludwig[full]'==$VERSION && \
pip cache purge

RUN apt-get purge -y build-essential cmake && apt-get -y autoremove && apt-get clean

0 comments on commit 70adf90

Please sign in to comment.