From b4d89b30948b84176d769a775b6d020a96751fb2 Mon Sep 17 00:00:00 2001
From: Xu Yang <peteryang@vip.qq.com>
Date: Mon, 15 Jul 2024 10:08:20 +0000
Subject: [PATCH] add a comment for GPU support

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

diff --git a/rdagent/scenarios/qlib/docker/Dockerfile b/rdagent/scenarios/qlib/docker/Dockerfile
index ff418d3a..26419c6e 100644
--- a/rdagent/scenarios/qlib/docker/Dockerfile
+++ b/rdagent/scenarios/qlib/docker/Dockerfile
@@ -1,4 +1,6 @@
-FROM pytorch/pytorch:latest  
+FROM pytorch/pytorch:latest
+
+# For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
 
 RUN apt-get clean && apt-get update && apt-get install -y \  
     curl \