diff --git a/rdagent/scenarios/qlib/docker/Dockerfile b/rdagent/scenarios/qlib/docker/Dockerfile index 3487f063..f8cb2ec0 100644 --- a/rdagent/scenarios/qlib/docker/Dockerfile +++ b/rdagent/scenarios/qlib/docker/Dockerfile @@ -1,4 +1,4 @@ -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