From 90cdcb7f0a366e9982a37593af6522975744190b Mon Sep 17 00:00:00 2001 From: "Jorge O. Castro" Date: Mon, 7 Aug 2023 18:35:25 -0400 Subject: [PATCH 1/2] feat(distrobox): add a premade config for pytorch-nvidia Co-authored-by: @lzumot --- usr/share/ublue-os/distrobox/pytorch-nvidia.ini | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 usr/share/ublue-os/distrobox/pytorch-nvidia.ini diff --git a/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/usr/share/ublue-os/distrobox/pytorch-nvidia.ini new file mode 100644 index 00000000000..79cb85730bb --- /dev/null +++ b/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 From 730f39caae21e48fb91f00010cf0cf8d32ee44bd Mon Sep 17 00:00:00 2001 From: "Jorge O. Castro" Date: Mon, 7 Aug 2023 18:47:01 -0500 Subject: [PATCH 2/2] fix: make this dx only --- {usr => dx/usr}/share/ublue-os/distrobox/pytorch-nvidia.ini | 0 1 file changed, 0 insertions(+), 0 deletions(-) rename {usr => dx/usr}/share/ublue-os/distrobox/pytorch-nvidia.ini (100%) diff --git a/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini similarity index 100% rename from usr/share/ublue-os/distrobox/pytorch-nvidia.ini rename to dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini