fix: pytorch-nvidia.ini to latest ngc and remove pre-init (#539)

ublue-os · Sep 30, 2023 · 4a8df6b · 4a8df6b
1 parent c0a561d
commit 4a8df6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini
@@ -1,8 +1,8 @@
 [mlbox]
-image=nvcr.io/nvidia/pytorch:23.08-py3
+image=nvcr.io/nvidia/pytorch:23.09-py3
 additional_packages="nano git htop"
 init_hooks="pip3 install huggingface_hub tokenizers transformers accelerate datasets wandb peft bitsandbytes fastcore fastprogress watermark torchmetrics deepspeed"
-pre-init-hooks="/init_script.sh"
+#pre-init-hooks="/init_script.sh"
 nvidia=true
 pull=true
 root=false