diff --git a/rdagent/scenarios/qlib/docker/Dockerfile b/rdagent/scenarios/qlib/docker/Dockerfile
index 9845c9a5..3487f063 100644
--- a/rdagent/scenarios/qlib/docker/Dockerfile
+++ b/rdagent/scenarios/qlib/docker/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch: latest
 
 # For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
 
@@ -20,4 +20,4 @@ RUN python -m pip install --upgrade cython
 RUN python -m pip install -e .
 
 RUN pip install catboost
-RUN pip install xgboost
\ No newline at end of file
+RUN pip install xgboost