BootStrap: docker
From: nvcr.io/nvidia/nemo:23.02

%environment
    export LC_ALL=C

%post
    cd /usr/local/lib/python3.8/dist-packages/nemo/collections/nlp/data/language_modeling/megatron
    make
    pip install accelerate