feat(distrobox): add a premade config for pytorch-nvidia (#385)

ublue-os · Aug 8, 2023 · 64a93dc · 64a93dc
1 parent 9a3b4b4
commit 64a93dc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini 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