Skip to content

Commit f7f0e54

Browse files
committed
aggregate 1/n
1 parent 5687ab2 commit f7f0e54

File tree

1 file changed

+51
-46
lines changed

1 file changed

+51
-46
lines changed

Dockerfile.tmpl

Lines changed: 51 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,12 @@ RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
106106
rm -rf /tmp/torch && \
107107
/tmp/clean-layer.sh
108108
{{ else }}
109-
RUN pip install torch==$TORCH_VERSION+cpu torchvision==$TORCHVISION_VERSION+cpu torchaudio==$TORCHAUDIO_VERSION+cpu torchtext==$TORCHTEXT_VERSION -f https://download.pytorch.org/whl/torch_stable.html && \
109+
RUN pip install \
110+
torch==$TORCH_VERSION+cpu \
111+
torchvision==$TORCHVISION_VERSION+cpu \
112+
torchaudio==$TORCHAUDIO_VERSION+cpu \
113+
torchtext==$TORCHTEXT_VERSION \
114+
-f https://download.pytorch.org/whl/torch_stable.html && \
110115
/tmp/clean-layer.sh
111116
{{ end }}
112117

@@ -155,39 +160,39 @@ RUN pip install spacy && \
155160
# Install GPU specific packages
156161
{{ if eq .Accelerator "gpu" }}
157162
# Install GPU-only packages
158-
RUN pip install pycuda && \
159-
pip install pynvrtc && \
160-
pip install pynvml && \
161-
pip install nnabla-ext-cuda$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \
163+
RUN pip install pycuda \
164+
pynvrtc \
165+
pynvml \
166+
nnabla-ext-cuda$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \
162167
/tmp/clean-layer.sh
163168
{{ end }}
164169

165-
RUN pip install pysal && \
166-
pip install seaborn python-dateutil dask python-igraph && \
167-
pip install pyyaml joblib husl geopy mne pyshp && \
168-
pip install pandas && \
169-
pip install flax && \
170+
RUN pip install pysal \
171+
seaborn python-dateutil dask python-igraph \
172+
pyyaml joblib husl geopy mne pyshp \
173+
pandas \
174+
flax \
170175
# Install h2o from source.
171176
# Use `conda install -c h2oai h2o` once Python 3.7 version is released to conda.
172177
apt-get install -y default-jre-headless && \
173-
pip install -f https://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o && \
174-
pip install tensorflow-gcs-config==2.6.0 && \
175-
pip install tensorflow-addons==0.14.0 && \
176-
pip install tensorflow_decision_forests==0.2.0 && \
178+
pip install -f https://h2o-release.s3.amazonaws.com/h2o/latest_stable_Py.html h2o \
179+
tensorflow-gcs-config==2.6.0 \
180+
tensorflow-addons==0.14.0 \
181+
tensorflow_decision_forests==0.2.0 && \
177182
/tmp/clean-layer.sh
178183

179184
RUN apt-get install -y libfreetype6-dev && \
180185
apt-get install -y libglib2.0-0 libxext6 libsm6 libxrender1 libfontconfig1 --fix-missing && \
181186
# b/198300835 kornia 4.1.0 is not compatible with our version of numpy.
182187
pip install gensim==4.0.1 && \
183-
pip install textblob && \
184-
pip install wordcloud && \
185-
pip install xgboost && \
186-
pip install pydot && \
187-
# Pinned because it breaks theano test with the latest version (b/178107003).
188-
pip install theano-pymc==1.0.11 && \
189-
pip install python-Levenshtein && \
190-
pip install hep_ml && \
188+
textblob \
189+
wordcloud \
190+
xgboost \
191+
pydot \
192+
# Pinned because it breaks theano test with the latest version (b/178107003).
193+
theano-pymc==1.0.11 \
194+
python-Levenshtein \
195+
hep_ml && \
191196
# NLTK Project datasets
192197
mkdir -p /usr/share/nltk_data && \
193198
# NLTK Downloader no longer continues smoothly after an error, so we explicitly list
@@ -208,8 +213,8 @@ RUN apt-get install -y libfreetype6-dev && \
208213
twitter_samples udhr2 udhr unicode_samples universal_tagset universal_treebanks_v20 \
209214
vader_lexicon verbnet webtext word2vec_sample wordnet wordnet_ic words ycoe && \
210215
# Stop-words
211-
pip install stop-words && \
212-
pip install scikit-image && \
216+
pip install stop-words \
217+
scikit-image && \
213218
/tmp/clean-layer.sh
214219

215220
RUN pip install ibis-framework && \
@@ -219,31 +224,31 @@ RUN pip install ibis-framework && \
219224
pip install gluoncv && \
220225
/tmp/clean-layer.sh
221226

222-
RUN pip install scipy && \
223-
pip install scikit-learn && \
224-
# Scikit-learn accelerated library for x86
225-
# TODO(b/262387811#4) Unpin when the package is fixed.
226-
pip install scikit-learn-intelex==2021.6.3 && \
227-
# HDF5 support
228-
pip install h5py && \
229-
pip install biopython && \
230-
# PUDB, for local debugging convenience
231-
pip install pudb && \
232-
pip install imbalanced-learn && \
233-
# Profiling and other utilities
234-
pip install line_profiler && \
235-
pip install orderedmultidict && \
236-
pip install smhasher && \
237-
pip install bokeh && \
238-
pip install numba && \
239-
pip install datashader && \
240-
# Boruta (python implementation)
241-
pip install Boruta && \
227+
RUN pip install scipy \
228+
scikit-learn \
229+
# Scikit-learn accelerated library for x86
230+
# TODO(b/262387811#4) Unpin when the package is fixed.
231+
scikit-learn-intelex==2021.6.3 \
232+
# HDF5 support
233+
h5py \
234+
biopython \
235+
# PUDB, for local debugging convenience
236+
pudb \
237+
imbalanced-learn \
238+
# Profiling and other utilities
239+
line_profiler \
240+
orderedmultidict \
241+
smhasher \
242+
bokeh \
243+
numba \
244+
datashader \
245+
# Boruta (python implementation)
246+
Boruta && \
242247
apt-get install -y graphviz && pip install graphviz && \
243248
# Pandoc is a dependency of deap
244249
apt-get install -y pandoc && \
245-
pip install git+https://github.com/scikit-learn-contrib/py-earth.git@issue191 && \
246-
pip install essentia && \
250+
pip install git+https://github.com/scikit-learn-contrib/py-earth.git@issue191 \
251+
essentia && \
247252
apt-get install -y git-lfs && \
248253
/tmp/clean-layer.sh
249254

0 commit comments

Comments
 (0)