Skip to content

Commit 6bb31b0

Browse files
committed
fix pytorch audio and drop allennlp and dlib
1 parent 3df3b6d commit 6bb31b0

File tree

4 files changed

+11
-32
lines changed

4 files changed

+11
-32
lines changed

Dockerfile.tmpl

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -402,7 +402,8 @@ RUN pip install bleach && \
402402
pip install ipykernel && \
403403
pip install ipython && \
404404
pip install ipython-genutils && \
405-
pip install ipywidgets && \
405+
# Fix qgrid by pinning ipywidgets https://github.com/quantopian/qgrid/issues/376
406+
pip install ipywidgets==7.7.1 && \
406407
pip install isoweek && \
407408
pip install jedi && \
408409
pip install jsonschema && \
@@ -445,7 +446,9 @@ RUN pip install bleach && \
445446
pip install pyarrow && \
446447
pip install feather-format && \
447448
pip install fastai && \
448-
pip install allennlp && \
449+
# allennlp is in maintenance mode https://github.com/allenai/allennlp
450+
# It downgrades the Pytorch install to 1.11.
451+
#pip install allennlp && \
449452
pip install importlib-metadata && \
450453
python -m spacy download en_core_web_sm && python -m spacy download en_core_web_lg && \
451454
apt-get install -y ffmpeg && \
@@ -504,7 +507,10 @@ RUN pip install flashtext && \
504507
pip install transformers && \
505508
# b/232247930 >= 2.2.0 requires pyarrow >= 6.0.0 which conflicts with dependencies for rapidsai 0.21.*
506509
pip install datasets==2.1.0 && \
507-
pip install dlib && \
510+
# dlib has a libmkl incompatibility:
511+
# test_dlib_face_detector (test_dlib.TestDLib) ... INTEL MKL ERROR: /opt/conda/bin/../lib/libmkl_avx512.so.2: undefined symbol: mkl_sparse_optimize_bsr_trsm_i8.
512+
# Intel MKL FATAL ERROR: Cannot load libmkl_avx512.so.2 or libmkl_def.so.2.
513+
#pip install dlib && \
508514
pip install kaggle-environments && \
509515
pip install geopandas && \
510516
pip install nnabla && \

packages/torch.Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,8 @@ ENV BUILD_VERSION=$TORCHAUDIO_VERSION
4545
RUN sudo apt-get update && \
4646
# ncurses.h is required for this install
4747
sudo apt-get install libncurses-dev && \
48+
# Fixing the build: https://github.com/pytorch/audio/issues/666#issuecomment-635928685
49+
conda install -c conda-forge ncurses && \
4850
cd /usr/local/src && \
4951
git clone https://github.com/pytorch/audio && \
5052
cd audio && \

tests/test_allennlp.py

Lines changed: 0 additions & 15 deletions
This file was deleted.

tests/test_dlib.py

Lines changed: 0 additions & 14 deletions
This file was deleted.

0 commit comments

Comments
 (0)