Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
inoue0426 authored Aug 30, 2024
1 parent 215dbbe commit 6e388a6
Showing 1 changed file with 39 additions and 0 deletions.
39 changes: 39 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,3 +41,42 @@ torch-spline-conv==1.2.2+pt113cu116

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 [email protected]:inoue0426/drGAT.git
cd drGAT
docker build -t drgat:latest .
docker run -it -p 9999:9999 drgat:latest
```

Then access to http://localhost:9999/notebooks/Tutorial.ipynb

## Installation using Conda

```shell
git clone [email protected]:inoue0426/drGAT.git
cd drGAT
conda env create -f environment.yml
conda activate drGAT
```
** NOTE: Please ensure the version matches exactly with your GPU/CPU specifications.


## Installation using requirement.txt

```shell
git clone [email protected]:inoue0426/drGAT.git
cd drGAT
conda create --name drGAT python=3.10 -y
conda activate drGAT
pip install -r requirement.txt
# Please make sure to change the version to match the version of your GPU/CPU machine exactly.
pip install --no-cache-dir torch==1.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install --no-cache-dir torch_geometric
pip install --no-cache-dir pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-1.13.1%2Bcu116.html
```
** NOTE: Please ensure the version matches exactly with your GPU/CPU specifications.

0 comments on commit 6e388a6

Please sign in to comment.