chore: bump pytorch version (ublue-os#671)

briorg · Dec 9, 2023 · f34d571 · f34d571
1 parent 3bc72cc
commit f34d571
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini
@@ -1,5 +1,5 @@
 [mlbox]
-image=nvcr.io/nvidia/pytorch:23.09-py3
+image=nvcr.io/nvidia/pytorch:23.10-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"