ajkndfjsdfasdf commited on
Commit
be85f98
·
verified ·
1 Parent(s): 01ee5d3

Upload 3 files

Browse files
Files changed (3) hide show
  1. Dockerfile +26 -0
  2. README.md +24 -0
  3. 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.")