| # Use Miniconda base image | |
| FROM continuumio/miniconda3 | |
| # Set working directory | |
| WORKDIR /app | |
| # Copy project files | |
| COPY . /app | |
| # Install system dependencies manually | |
| RUN apt-get update && apt-get install -y \ | |
| libstdc++6 \ | |
| libgcc-s1 \ | |
| libgomp1 \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Install Mamba for faster dependency resolution | |
| RUN conda install -n base -c conda-forge mamba | |
| # Create Conda environment using Mamba | |
| RUN mamba env create -f environment.yml | |
| # Activate Conda environment | |
| SHELL ["conda", "run", "-n", "depth-jitter", "/bin/bash", "-c"] | |
| # Install any additional dependencies | |
| RUN conda run -n depth-jitter pip install --no-cache-dir torch torchvision pytorch-lightning | |
| # Expose Jupyter port (if needed) | |
| EXPOSE 8888 | |
| # Allow argument passing | |
| ENTRYPOINT ["conda", "run", "-n", "depth-jitter", "python", "train_q2l.py"] | |