| # 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 |