# Use the NVIDIA CUDA image as the base image
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

# Install dependencies
RUN apt-get update && apt-get install -y wget git

# Download and install Miniconda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
    bash Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \
    rm Miniconda3-latest-Linux-x86_64.sh

# Add conda to PATH
ENV PATH /opt/conda/bin:$PATH