Update Dockerfile
Browse files- Dockerfile +4 -0
Dockerfile
CHANGED
|
@@ -4,6 +4,8 @@ FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
|
|
| 4 |
# 2. Avoid interactive timezone/keyboard prompts during apt installation
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 6 |
|
|
|
|
|
|
|
| 7 |
# 3. Install core system dependencies required by Unsloth's install.sh
|
| 8 |
RUN apt-get update && apt-get install -y \
|
| 9 |
curl \
|
|
@@ -29,6 +31,8 @@ ENV HF_HOME=/data/huggingface
|
|
| 29 |
|
| 30 |
WORKDIR $HOME
|
| 31 |
|
|
|
|
|
|
|
| 32 |
# 6. Execute the official Unsloth installation script.
|
| 33 |
# This will download PyTorch, Unsloth, Node/Bun, and compile the frontend during the Docker build.
|
| 34 |
RUN curl -fsSL https://unsloth.ai/install.sh | sh
|
|
|
|
| 4 |
# 2. Avoid interactive timezone/keyboard prompts during apt installation
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 6 |
|
| 7 |
+
|
| 8 |
+
|
| 9 |
# 3. Install core system dependencies required by Unsloth's install.sh
|
| 10 |
RUN apt-get update && apt-get install -y \
|
| 11 |
curl \
|
|
|
|
| 31 |
|
| 32 |
WORKDIR $HOME
|
| 33 |
|
| 34 |
+
USER root
|
| 35 |
+
|
| 36 |
# 6. Execute the official Unsloth installation script.
|
| 37 |
# This will download PyTorch, Unsloth, Node/Bun, and compile the frontend during the Docker build.
|
| 38 |
RUN curl -fsSL https://unsloth.ai/install.sh | sh
|