diff --git a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini
index d54df11535a..eb7ee2e1d46 100644
--- a/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini
+++ b/dx/usr/share/ublue-os/distrobox/pytorch-nvidia.ini
@@ -1,7 +1,7 @@
 [mlbox]
-image=nvcr.io/nvidia/pytorch:23.07-py3
+image=nvcr.io/nvidia/pytorch:23.08-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"
+init_hooks="pip3 install huggingface_hub tokenizers transformers accelerate datasets wandb peft bitsandbytes fastcore fastprogress watermark torchmetrics deepspeed"
 pre-init-hooks="/init_script.sh"
 nvidia=true
 pull=true