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

awesomekyle · Aug 8, 2023 · b15c905 · b15c905
1 parent 4b61087
commit b15c905
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