Docker

Dockerfile

@@ -42,11 +42,9 @@ RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/privacy-2.0.8-p
 
 RUN pip install --no-cache-dir /Responsible-Ai-Moderation-Models/en_core_web_lg-3.5.0-py3-none-any.whl
 
-# Uninstall any existing PyTorch versions
-RUN pip uninstall -y torch torchvision torchaudio
+RUN pip install torch==2.2.0+cu118 torchaudio==2.2.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
 
-# Install the desired version from the PyTorch CUDA 11.8 index
-RUN pip install torch==2.2.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.2.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+RUN pip install torchvision --extra-index-url https://download.pytorch.org/whl/cu118
 
 
 # RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118