Upload Dockerfile.dockerfile

Dockerfile.dockerfile

@@ -0,0 +1,23 @@
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+
+# Install Python and pip
+RUN apt-get update && apt-get install -y python3 python3-pip git
+
+# Set python3 as default
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
+# Set working directory
+WORKDIR /app
+
+# Copy local files to container
+COPY . .
+
+# Upgrade pip and install dependencies
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
+
+# (Optional) Set Hugging Face token — or use Secrets instead
+ENV HF_TOKEN="your_token_here"
+
+# Run your script when container starts
+CMD ["python", "train.py"]