diff --git a/Dockerfile_amd b/Dockerfile_amd
index c2ec4a6d..9a5e3568 100644
--- a/Dockerfile_amd
+++ b/Dockerfile_amd
@@ -36,7 +36,7 @@ COPY launcher launcher
 RUN cargo build --release
 
 # Text Generation Inference base image for RoCm
-FROM rocm/dev-ubuntu-20.04:5.7 as base
+FROM rocm/dev-ubuntu-22.04:5.7 as base
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     build-essential \