File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -27,6 +27,8 @@ RUN apt-get update && \
2727
2828ENV _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
3133RUN 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
9496ARG TARGETPLATFORM
9597
9698COPY . /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/
98100RUN 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
107109FROM builder as deploy
108110COPY . /modulus/
109- RUN cd /modulus/ && pip install .[all]
111+ RUN cd /modulus/ && pip install .
110112RUN pip install "protobuf==3.20.3"
111113
112114# Clean up
You can’t perform that action at this time.
0 commit comments