From 89afe914e84cc9a06d1074490451fb224255a770 Mon Sep 17 00:00:00 2001
From: Xisen Wang <118058822+Xisen-Wang@users.noreply.github.com>
Date: Fri, 19 Jul 2024 14:59:45 +0800
Subject: [PATCH] Update Dockerfile

---
 rdagent/scenarios/qlib/docker/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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