Upload 3 files
Browse files- Dockerfile +26 -0
- README.md +24 -0
- train.py +1 -0
Dockerfile
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
|
| 2 |
+
|
| 3 |
+
RUN apt-get update && apt-get install -y \
|
| 4 |
+
git \
|
| 5 |
+
git-lfs \
|
| 6 |
+
screen \
|
| 7 |
+
curl \
|
| 8 |
+
&& git lfs install \
|
| 9 |
+
&& apt-get clean
|
| 10 |
+
|
| 11 |
+
RUN pip install --upgrade pip
|
| 12 |
+
RUN pip install \
|
| 13 |
+
torch \
|
| 14 |
+
transformers \
|
| 15 |
+
datasets \
|
| 16 |
+
accelerate \
|
| 17 |
+
scipy \
|
| 18 |
+
safetensors \
|
| 19 |
+
sentencepiece \
|
| 20 |
+
wandb
|
| 21 |
+
|
| 22 |
+
WORKDIR /workspace
|
| 23 |
+
|
| 24 |
+
COPY . .
|
| 25 |
+
|
| 26 |
+
CMD echo "✅ Контейнер готов. Зайди в /workspace и запускай train.py вручную."
|
README.md
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Custom Docker for RunPod / Hugging Face
|
| 2 |
+
|
| 3 |
+
This archive includes:
|
| 4 |
+
- Dockerfile (based on `pytorch:2.1.0-cuda11.8`)
|
| 5 |
+
- Minimal `train.py`
|
| 6 |
+
- Required Python packages
|
| 7 |
+
|
| 8 |
+
## Usage
|
| 9 |
+
|
| 10 |
+
Upload to Hugging Face 🤗 Hub and use it as a custom Docker template in RunPod or similar platforms.
|
| 11 |
+
|
| 12 |
+
### RunPod Setup
|
| 13 |
+
|
| 14 |
+
1. Container Type: **Dockerfile from GitHub/URL**
|
| 15 |
+
2. GitHub or HF repo: Your hosted archive
|
| 16 |
+
3. Dockerfile Path: `Dockerfile`
|
| 17 |
+
4. Start Command: `bash`
|
| 18 |
+
|
| 19 |
+
Then connect and run:
|
| 20 |
+
```bash
|
| 21 |
+
cd /workspace && python train.py
|
| 22 |
+
```
|
| 23 |
+
|
| 24 |
+
✅ Happy training!
|
train.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
print("✅ Training script is ready. Customize this file as needed.")
|