flair messes with pytorch?

fhswf · Aug 29, 2024 · 29609d5 · 29609d5
1 parent 06d8e82
commit 29609d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/images/deepml/Dockerfile b/images/deepml/Dockerfile
@@ -59,12 +59,13 @@ ENV TF_FORCE_GPU_ALLOW_GROWTH true
 RUN pip install \
     torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 \
     #torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117 \
-    && rm -rf /tmp/* $HOME/.cache
+
+RUN pip install flair no-dependencies && rm -rf /tmp/* $HOME/.cache
 
 # Install Huggingface tensorflow and other libraries
 RUN pip install \
     nvidia-cudnn-cu12==9.2.1.18 tensorflow==2.17.* mysql-connector-python keras-tuner pymysql \
-    librosa gradio wandb Cython==0.29.36 flair \ 
+    librosa gradio wandb Cython==0.29.36 \ 
     transformers datasets tokenizers evaluate adapter-transformers diffusers[torch] accelerate \
     nltk gensim jiwer audiofile opencv-python tiktoken \
     && rm -rf /tmp/* $HOME/.cache