From 64a93dc7636bfc7fee03fa5cb8e6501ffeb3bfcc Mon Sep 17 00:00:00 2001
From: "Jorge O. Castro" <jorge.castro@gmail.com>
Date: Tue, 8 Aug 2023 10:49:48 -0500
Subject: [PATCH] feat(distrobox): add a premade config for pytorch-nvidia
 (#385)

---
 dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini

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