Skip to content

Commit 450c67d

Browse files
authored
update dependency installs in dockerfile (#164)
1 parent 6298feb commit 450c67d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@ RUN apt-get update && \
2727

2828
ENV _CUDA_COMPAT_TIMEOUT=90
2929

30+
# Install other dependencies
31+
RUN pip install "h5py>=3.7.0" "mpi4py>=3.1.4" "netcdf4>=1.6.3" "ruamel.yaml>=0.17.22" "scikit-learn>=1.0.2"
3032
# TODO remove benchy dependency
3133
RUN pip install git+https://github.com/romerojosh/benchy.git
3234
# TODO use torch-harmonics pip package after the upgrade
@@ -94,7 +96,7 @@ FROM builder as ci
9496
ARG TARGETPLATFORM
9597

9698
COPY . /modulus/
97-
RUN cd /modulus/ && pip install -e .[all] && pip uninstall nvidia-modulus -y && rm -rf /modulus/
99+
RUN cd /modulus/ && pip install -e . && pip uninstall nvidia-modulus -y && rm -rf /modulus/
98100
RUN if [ "$TARGETPLATFORM" = "linux/amd64" ]; then \
99101
echo "Installing tensorflow and warp-lang for: $TARGETPLATFORM" && \
100102
pip install "tensorflow==2.9.0" "warp-lang>=0.6.0"; \
@@ -106,7 +108,7 @@ RUN pip install "black==22.10.0" "interrogate==1.5.0" "coverage==6.5.0" "protobu
106108
# Deployment image
107109
FROM builder as deploy
108110
COPY . /modulus/
109-
RUN cd /modulus/ && pip install .[all]
111+
RUN cd /modulus/ && pip install .
110112
RUN pip install "protobuf==3.20.3"
111113

112114
# Clean up

0 commit comments

Comments
 (0)