<html><head><meta name="color-scheme" content="light dark"></head><body><pre style="word-wrap: break-word; white-space: pre-wrap;">Bootstrap: docker

From: pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime

%post

    apt-get update &amp;&amp; apt-get install -y --no-install-recommends \
        software-properties-common \
        build-essential \
        curl \
        git \
        ffmpeg \
        texlive-latex-extra \
        texlive-fonts-recommended \
        texlive-base \
        dvipng

    conda create -n ffcv python=3.9 \
        cupy \
        pkg-config \
        compilers \
        libjpeg-turbo \
        opencv \
        pytorch \
        torchvision \
        ipykernel \
        requests \
        imageio \
        scikit-image \
        einops \
        vit-pytorch \
        cudatoolkit=11.3 \
        numba -c pytorch -c conda-forge

    echo "source activate" &gt;&gt; ~/.bashrc
    echo "conda activate ffcv" &gt;&gt; ~/.bashrc

    git clone https://github.com/libffcv/ffcv.git

    conda run -n ffcv pip install ffcv


%runscript
    exec "$@"


</pre></body></html>