From 0798b508015b8676f3d830963c59754b5d02c406 Mon Sep 17 00:00:00 2001
From: Yoshitaka Inoue <8393063+inoue0426@users.noreply.github.com>
Date: Mon, 8 Jul 2024 11:20:58 -0400
Subject: [PATCH] Update README.md

---
 README.md | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 6a67343..954f37f 100644
--- a/README.md
+++ b/README.md
@@ -36,7 +36,19 @@ torch-spline-conv==1.2.2+pt113cu116
 
 ## Environment
 
-Our experiment was conducted on Ubuntu with an NVIDIA A100 Tensor Core GPU.
+Our experiment was conducted on Ubuntu with an NVIDIA A100 Tensor Core GPU.  
+If you want to re-train model, we reccomend to use GPU.
+
+## Installation using Docker
+
+```shell
+git clone git@github.com:inoue0426/drGAT.git
+cd drGAT
+docker run -it -p 9999:9999 drgat:latest
+docker build -t drgat:latest .
+```
+
+Then access to http://localhost:9999/notebooks/Tutorial.ipynb 
 
 ## Installation using Conda