[assets] fix docker images (#8203)

docker/docker-cuda/Dockerfile

@@ -1,5 +1,5 @@
 # https://hub.docker.com/r/hiyouga/pytorch/tags
-ARG BASE_IMAGE=hiyouga/pytorch:th2.6.0-cu124-flashattn2.7.4-cxx11abi0
+ARG BASE_IMAGE=hiyouga/pytorch:th2.6.0-cu124-flashattn2.7.4-cxx11abi0-devel
 FROM ${BASE_IMAGE}
 
 # Installation arguments
@@ -47,7 +47,7 @@ RUN if [ "${INSTALL_FLASHATTN}" == "true" ]; then \
     fi
 
 # Set up volumes
-VOLUME [ "/root/.cache/huggingface", "/root/.cache/modelscope", "/root/.cache/openmind", "/app/data", "/app/output" ]
+VOLUME [ "/root/.cache/huggingface", "/root/.cache/modelscope", "/root/.cache/openmind", "/app/shared_data", "/app/output" ]
 
 # Expose port 7860 for LLaMA Board
 ENV GRADIO_SERVER_PORT=7860

docker/docker-cuda/Dockerfile.base

@@ -1,6 +1,6 @@
 # Start from the pytorch official image (ubuntu-22.04 + cuda-12.4.1 + python-3.11)
 # https://hub.docker.com/r/pytorch/pytorch/tags
-FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
 
 # Define environments
 ENV MAX_JOBS=16

docker/docker-cuda/docker-compose.yml

@@ -11,7 +11,7 @@ services:
       - ../../hf_cache:/root/.cache/huggingface
       - ../../ms_cache:/root/.cache/modelscope
       - ../../om_cache:/root/.cache/openmind
-      - ../../data:/app/data
+      - ../../shared_data:/app/shared_data
       - ../../output:/app/output
     ports:
       - "7860:7860"