UL2-nemo-conversion / nemo_singularity.def
Faton Rekathati
UL2 conversion instructions
0fd282e
raw
history blame
230 Bytes
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