From 57e0bc35d1be85d5c5b3009d9b5bd561b31725f3 Mon Sep 17 00:00:00 2001 From: Peter Knauer Date: Sun, 7 Dec 2025 18:43:24 -0500 Subject: [PATCH] v0.1 - initial commit --- Dockerfile | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 8e40cc9..870f3d4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,4 @@ # ROCm-enabled PyTorch base image from AMD -# Pick a tag that matches your installed ROCm driver best; this one is a good default. FROM rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.10_pytorch_release_2.9.1 # OCI metadata for discoverability @@ -16,11 +15,10 @@ RUN apt-get update && \ rm -rf /var/lib/apt/lists/* # Upgrade pip and install Unsloth (AMD is supported via regular pip install) -# Unsloth itself is Apache-2.0 licensed. RUN pip install --upgrade pip && \ pip install --no-cache-dir unsloth -# Optional: tools you’ll almost certainly want when experimenting +# Optional tools for experimenting RUN pip install --no-cache-dir \ "transformers>=4.45.0" \ datasets \ @@ -41,4 +39,4 @@ USER ${USERNAME} WORKDIR /workspace # Keep the container alive by default; you’ll exec into it -CMD ["bash"] +CMD ["bash"] \ No newline at end of file