diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm
index 5e3f756dcd35833132ccff071139e0e7769cf4ab..cdfe5e85db4e1472027830d81707ff7772571dc2 100644
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -1,4 +1,4 @@
-FROM rocm/pytorch:rocm5.0.1_ubuntu18.04_py3.7_pytorch_1.10.0
+FROM rocm/pytorch:latest
 
 # install latest released version of deepspeed
 RUN pip install deepspeed && \