simple Dockerfile for rurnning mlx

[AlexCheema]

No description provided.

[thenatlog]

LGTM

[johnykes]

Just tried this on both WSL and MacOS

MacOS

ModuleNotFoundError: No module named 'mlx_lm.models.cahe'

WSL - ubuntu 22

29.52 [ 68%] Built target mlx
29.53 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'examples/cpp/tutorial'.  Stop.
29.53 make[2]: *** Waiting for unfinished jobs....
29.53 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'examples/cpp/distributed'.  Stop.
29.53 make[2]: *** Waiting for unfinished jobs....
29.54 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'examples/cpp/logistic_regression'.  Stop.
29.54 make[2]: *** Waiting for unfinished jobs....
29.54 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'examples/cpp/metal_capture'.  Stop.
29.54 make[2]: *** Waiting for unfinished jobs....
29.54 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'examples/cpp/linear_regression'.  Stop.
29.54 make[2]: *** Waiting for unfinished jobs....
29.54 [ 69%] Building CXX object examples/cpp/CMakeFiles/tutorial.dir/tutorial.cpp.o
29.54 [ 70%] Building CXX object tests/CMakeFiles/tests.dir/tests.cpp.o
29.54 [ 71%] Building CXX object tests/CMakeFiles/tests.dir/array_tests.cpp.o
29.54 [ 72%] Building CXX object examples/cpp/CMakeFiles/distributed.dir/distributed.cpp.o
29.54 [ 73%] Building CXX object tests/CMakeFiles/tests.dir/allocator_tests.cpp.o
29.54 [ 74%] Building CXX object examples/cpp/CMakeFiles/metal_capture.dir/metal_capture.cpp.o
29.54 [ 75%] Building CXX object examples/cpp/CMakeFiles/logistic_regression.dir/logistic_regression.cpp.o
29.54 [ 76%] Building CXX object tests/CMakeFiles/tests.dir/arg_reduce_tests.cpp.o
29.54 [ 77%] Building CXX object tests/CMakeFiles/tests.dir/einsum_tests.cpp.o
29.54 [ 78%] Building CXX object examples/cpp/CMakeFiles/linear_regression.dir/linear_regression.cpp.o
29.54 [ 79%] Building CXX object tests/CMakeFiles/tests.dir/creations_tests.cpp.o
29.55 [ 80%] Building CXX object tests/CMakeFiles/tests.dir/compile_tests.cpp.o
29.55 [ 81%] Building CXX object tests/CMakeFiles/tests.dir/custom_vjp_tests.cpp.o
29.55 [ 82%] Building CXX object tests/CMakeFiles/tests.dir/device_tests.cpp.o
29.55 [ 83%] Building CXX object tests/CMakeFiles/tests.dir/blas_tests.cpp.o
29.55 [ 84%] Building CXX object tests/CMakeFiles/tests.dir/load_tests.cpp.o
29.55 [ 85%] Building CXX object tests/CMakeFiles/tests.dir/autograd_tests.cpp.o
29.55 [ 87%] Building CXX object tests/CMakeFiles/tests.dir/eval_tests.cpp.o
29.55 [ 87%] Building CXX object tests/CMakeFiles/tests.dir/fft_tests.cpp.o
29.55 [ 88%] Building CXX object tests/CMakeFiles/tests.dir/ops_tests.cpp.o
29.59 [ 89%] Building CXX object tests/CMakeFiles/tests.dir/scheduler_tests.cpp.o
29.60 [ 91%] Building CXX object tests/CMakeFiles/tests.dir/vmap_tests.cpp.o
29.60 [ 91%] Building CXX object tests/CMakeFiles/tests.dir/random_tests.cpp.o
29.62 [ 92%] Building CXX object tests/CMakeFiles/tests.dir/utils_tests.cpp.o
29.63 make[2]: *** No rule to make target '/usr/lib/aarch64-linux-gnu/liblapack.so', needed by 'tests/tests'.  Stop.
29.63 make[2]: *** Waiting for unfinished jobs....
29.66 [ 93%] Building CXX object tests/CMakeFiles/tests.dir/linalg_tests.cpp.o
30.93 make[1]: *** [CMakeFiles/Makefile2:1288: examples/cpp/CMakeFiles/distributed.dir/all] Error 2
30.93 make[1]: *** Waiting for unfinished jobs....
31.18 make[1]: *** [CMakeFiles/Makefile2:1261: examples/cpp/CMakeFiles/metal_capture.dir/all] Error 2
31.69 make[1]: *** [CMakeFiles/Makefile2:1180: examples/cpp/CMakeFiles/tutorial.dir/all] Error 2
31.99 make[1]: *** [CMakeFiles/Makefile2:1207: examples/cpp/CMakeFiles/linear_regression.dir/all] Error 2
32.28 make[1]: *** [CMakeFiles/Makefile2:1234: examples/cpp/CMakeFiles/logistic_regression.dir/all] Error 2
35.09 make[1]: *** [CMakeFiles/Makefile2:1127: tests/CMakeFiles/tests.dir/all] Error 2
35.09 make: *** [Makefile:146: all] Error 2
------
failed to solve: process "/bin/sh -c 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 ." did not complete successfully: exit code: 2