From 42bec9fa6cf57981246165c46f9140e4858c0cad Mon Sep 17 00:00:00 2001 From: "Alexander.Kang" Date: Wed, 2 Oct 2024 03:00:12 +1000 Subject: [PATCH] Update README.md again --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index aeb33d7..1e2b065 100644 --- a/README.md +++ b/README.md @@ -15,6 +15,9 @@ Below is an overview of our EGNN model architecture: ## Environment Setup +## Environment Setup +The code is tested on pyg (Pytorch-Geometric) **2.4.0**, **python 3.8**, **Pytorch 2.0.0**, **cuda11.8**, **GPU RAM** at least 8GB with batch size 1 on **GTX 2080** above. + To set up the environment for this project, we use Conda. Follow these steps: 1. Make sure you have Conda installed. If not, you can download it from [here](https://docs.conda.io/en/latest/miniconda.html).