Forráskód Böngészése

simple Dockerfile for rurnning mlx

Alex Cheema 1 éve
szülő
commit
341a9d29e7
1 módosított fájl, 35 hozzáadás és 0 törlés
  1. 35 0
      Dockerfile

+ 35 - 0
Dockerfile

@@ -0,0 +1,35 @@
+FROM python:3.12-slim
+
+RUN apt-get update && apt-get install -y \
+    software-properties-common \
+    build-essential \
+    pkg-config \
+    cmake \
+    libopenblas-dev \
+    liblapack-dev \
+    liblapacke-dev \
+    python3-pip \
+    curl \
+    git
+
+RUN git clone https://github.com/ml-explore/mlx.git && cd mlx && mkdir -p build && cd build && \
+    cmake .. \
+      -DCMAKE_PREFIX_PATH="/usr/lib/aarch64-linux-gnu" \
+      -DLAPACK_LIBRARIES="/usr/lib/aarch64-linux-gnu/liblapack.so" \
+      -DBLAS_LIBRARIES="/usr/lib/aarch64-linux-gnu/libopenblas.so" \
+      -DLAPACK_INCLUDE_DIRS="/usr/include" && \
+    sed -i 's/option(MLX_BUILD_METAL "Build metal backend" ON)/option(MLX_BUILD_METAL "Build metal backend" OFF)/' ../CMakeLists.txt && \
+    make -j && \
+    make install && \
+    cd .. && \
+    pip install --no-cache-dir . && \
+    rm -rf build mlx
+
+RUN pip install --no-cache-dir numpy
+
+# Add library path
+# ENV LD_LIBRARY_PATH="/usr/local/lib:${LD_LIBRARY_PATH}"
+
+COPY test.py .
+
+CMD ["python3", "test.py"]