diff --git a/docker/Dockerfile.remyx b/docker/Dockerfile.remyx new file mode 100644 index 00000000..43ab2489 --- /dev/null +++ b/docker/Dockerfile.remyx @@ -0,0 +1,28 @@ +# syntax=docker/dockerfile:1.4 + +# Base Image: No torch version pinned, using default `pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel` +FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel + +# Set CUDA_HOME environment variable to match the base image +ENV CUDA_HOME=/usr/local/cuda-12.1 +ENV DEBIAN_FRONTEND=noninteractive + +# Set the working directory +WORKDIR /app + +# Install git, needed for dependencies specified with git+https URLs +RUN apt-get update && \ + apt-get install -y --no-install-recommends git && \ + rm -rf /var/lib/apt/lists/* + +# Clone the repository into the working directory +RUN git clone --depth 1 https://github.com/automl/ConfigurableOptimizer . + +# Install the package and its dependencies. +# The package is installed in editable mode (-e) so that scripts in 'examples/' +# can import the 'confopt' module from 'src/'. This will also install +# all dependencies from pyproject.toml including tensorflow via nasbench. +RUN pip install --no-cache-dir -e . + +# Specify the command to run the demo +CMD ["python", "examples/demo_light.py"] \ No newline at end of file