diff --git a/deepmd-kit/Dockerfile b/deepmd-kit/Dockerfile
index 95bed61..ac61a80 100644
--- a/deepmd-kit/Dockerfile
+++ b/deepmd-kit/Dockerfile
@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=ubuntu:focal
+ARG BASE_IMAGE=almalinux:8
 FROM $BASE_IMAGE
 
 LABEL maintainer="Jinzhe Zeng <jinzhe.zeng@rutgers.edu>" \