@@ -106,7 +106,12 @@ RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
106
106
rm -rf /tmp/torch && \
107
107
/tmp/clean-layer.sh
108
108
{{ 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 && \
110
115
/tmp/clean-layer.sh
111
116
{{ end }}
112
117
@@ -155,39 +160,39 @@ RUN pip install spacy && \
155
160
# Install GPU specific packages
156
161
{{ if eq .Accelerator "gpu" }}
157
162
# 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 && \
162
167
/tmp/clean-layer.sh
163
168
{{ end }}
164
169
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 \
170
175
# Install h2o from source.
171
176
# Use `conda install -c h2oai h2o` once Python 3.7 version is released to conda.
172
177
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 && \
177
182
/tmp/clean-layer.sh
178
183
179
184
RUN apt-get install -y libfreetype6-dev && \
180
185
apt-get install -y libglib2.0-0 libxext6 libsm6 libxrender1 libfontconfig1 --fix-missing && \
181
186
# b/198300835 kornia 4.1.0 is not compatible with our version of numpy.
182
187
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 && \
191
196
# NLTK Project datasets
192
197
mkdir -p /usr/share/nltk_data && \
193
198
# NLTK Downloader no longer continues smoothly after an error, so we explicitly list
@@ -208,8 +213,8 @@ RUN apt-get install -y libfreetype6-dev && \
208
213
twitter_samples udhr2 udhr unicode_samples universal_tagset universal_treebanks_v20 \
209
214
vader_lexicon verbnet webtext word2vec_sample wordnet wordnet_ic words ycoe && \
210
215
# Stop-words
211
- pip install stop-words && \
212
- pip install scikit-image && \
216
+ pip install stop-words \
217
+ scikit-image && \
213
218
/tmp/clean-layer.sh
214
219
215
220
RUN pip install ibis-framework && \
@@ -219,31 +224,31 @@ RUN pip install ibis-framework && \
219
224
pip install gluoncv && \
220
225
/tmp/clean-layer.sh
221
226
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 && \
242
247
apt-get install -y graphviz && pip install graphviz && \
243
248
# Pandoc is a dependency of deap
244
249
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 && \
247
252
apt-get install -y git-lfs && \
248
253
/tmp/clean-layer.sh
249
254
0 commit comments