diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini new file mode 100644 index 00000000000..79cb85730bb --- /dev/null +++ b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini @@ -0,0 +1,9 @@ +[mlbox] +image=nvcr.io/nvidia/pytorch:23.07-py3 +additional_packages="nano git htop" +init_hooks="pip3 install huggingface_hub tokenizers transformers accelerate datasets wandb peft bitsandbytes fastcore fastprogress watermark torchmetrics deepspeed xformers" +pre-init-hooks="/init_script.sh" +nvidia=true +pull=true +root=false +replace=true