diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini index e9cf02a8d4b..c9de08221db 100644 --- 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"