build with no libraries and pass in the device via compose
This commit is contained in:
@@ -1,19 +1,15 @@
|
||||
# Use the official ROCm base image
|
||||
FROM rocm/dev-ubuntu-24.04:6.2
|
||||
FROM ubuntu:24.04
|
||||
|
||||
# Set environment variables for 7900 XTX (gfx1100)
|
||||
ENV HSA_OVERRIDE_GFX_VERSION=11.0.0
|
||||
ENV TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
|
||||
ENV PYTHONUNBUFFERED=1
|
||||
|
||||
# Install system dependencies
|
||||
RUN apk add --no-cache \
|
||||
python3 \
|
||||
py3-pip \
|
||||
git \
|
||||
mesa-gles \
|
||||
glib \
|
||||
&& rm -rf /var/cache/apk/*
|
||||
# Install basic Python and Graphics libs
|
||||
RUN apt update -y && apt install -y \
|
||||
python3-pip python3-dev git libgl1-mesa-glx libglib2.0-0 \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /app
|
||||
|
||||
Reference in New Issue
Block a user