diff --git a/Dockerfile.worker.gguf b/Dockerfile.worker.gguf
index 114b584..535ae71 100644
--- a/Dockerfile.worker.gguf
+++ b/Dockerfile.worker.gguf
@@ -1,4 +1,4 @@
-FROM docker.io/rocm/pytorch:rocm6.3.2_ubuntu24.04_py3.12
+FROM docker.io/rocm/pytorch:rocm6.4.4_ubuntu24.04_py3.12_pytorch_release_2.6.0
 
 WORKDIR /app