27 lines
776 B
Docker
27 lines
776 B
Docker
# Unsloth Studio on AMD GPU (780M)
|
|
FROM rocm/pytorch:rocm6.4.2_ubuntu24.04_py3.12_pytorch_release_2.4.1
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && \
|
|
apt-get install -y curl git wget ffmpeg libglvnd libegl-mesa0 libdrm2
|
|
|
|
# Install Unsloth Studio with RoCM support
|
|
RUN pip install --upgrade pip && \
|
|
pip install unsloth[studio] torch>=2.4.1 && \
|
|
rm -rf /root/.cache/pip
|
|
|
|
# Set working directory
|
|
WORKDIR /app
|
|
|
|
# Copy application files
|
|
COPY . /app/
|
|
|
|
# Create a volume for persistent storage (weights, checkpoints)
|
|
VOLUME ["/data"]
|
|
|
|
# Environment variables for AMD GPU compatibility
|
|
ENV UNSLOTH_ALLOW_XLA=1 \
|
|
PYTORCH_CUDA_ALLOC_FAIL=0
|
|
|
|
# Run Unsloth Studio with the model path
|
|
CMD ["unsloth", "studio", "--cpu", "--gpu", "0", "--model", "/data/my-model"]
|