fix

docker/docker-npu/Dockerfile

@@ -40,6 +40,9 @@ RUN pip uninstall -y torch torchvision torchaudio && \
 # Set up volumes
 # VOLUME [ "/root/.cache/huggingface", "/app/shared_data", "/app/output" ]
 
+# Verify torch version
+RUN python -c "import torch, sys; v=torch.__version__; print('torch version:', v); sys.exit(0 if v.startswith('2.7.1') else 1)"
+
 # Expose port 7860 for LLaMA Board
 ENV GRADIO_SERVER_PORT=7860
 EXPOSE 7860

docker/docker-npu/docker-compose.yml

@@ -35,7 +35,6 @@ services:
       args:
         BASE_IMAGE: quay.io/ascend/cann:8.3.rc2-a3-ubuntu22.04-py3.11
         PIP_INDEX: https://pypi.org/simple
-        EXTRAS: torch-npu,metrics
     container_name: llamafactory-a3
     image: llamafactory:npu-a3
     volumes: