From ad5ea1cf23449d1cc75154f9c20c85322141eeaf Mon Sep 17 00:00:00 2001
From: Lizzyhku <72957393+lizzyhku@users.noreply.github.com>
Date: Wed, 22 Feb 2023 10:27:56 +0800
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 97bf315..2293463 100644
--- a/README.md
+++ b/README.md
@@ -4,7 +4,7 @@
 This is the implementation of Automated Spatio-Temporal Graph Contrastive Learning (AutoST):
 
 ## Requirements ##
-Pytorch = 1.7.0 and Tensorflow = 1.15.3 (crime prediction task (ST-SHN))
+python = 3.7.13, Pytorch = 1.12.0 with cuda = 10.2  and Tensorflow = 1.15.3 (crime prediction task (ST-SHN))
 
 ## Data ##
 Moblity data, crime data, census block data, POI data, house price of Chicago ([https://data.cityofchicago.org/](https://data.cityofchicago.org/)) and New York([https://opendata.cityofnewyork.us/](https://opendata.cityofnewyork.us/)).