初始化项目,由ModelHub XC社区提供模型
Model: jiamingshan/AHA-L2A-Qwen3-1.7B-repro Source: Original Platform
This commit is contained in:
16
recipe/Dockerfile
Normal file
16
recipe/Dockerfile
Normal file
@@ -0,0 +1,16 @@
|
||||
FROM pytorch/pytorch:2.9.1-cuda12.8-cudnn9-runtime
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive \
|
||||
PYTHONUNBUFFERED=1 \
|
||||
TOKENIZERS_PARALLELISM=false \
|
||||
HF_HOME=/workspace/.cache/huggingface \
|
||||
PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
||||
|
||||
RUN apt-get update \
|
||||
&& apt-get install -y --no-install-recommends git ca-certificates \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
COPY requirements.txt /tmp/requirements.txt
|
||||
RUN python -m pip install --no-cache-dir -r /tmp/requirements.txt
|
||||
|
||||
WORKDIR /workspace/recipe
|
||||
Reference in New Issue
Block a user