Skip to content

Commit

Permalink
[Doc] Update Readme for example Hilander (dmlc#4587)
Browse files Browse the repository at this point in the history
The example was updated in dmlc#3972 but the README file was not. This PR updates it to reflect the correct environment requirement.

Co-authored-by: Minjie Wang <[email protected]>
  • Loading branch information
hetong007 and jermainewang authored Sep 22, 2022
1 parent 92bff4d commit 33c8abb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/pytorch/hilander/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,13 @@ This folder contains the official code for [Learning Hierarchical Graph Neural N

## Setup

We use python 3.7. The CUDA version needs to be 10.2. Besides DGL (>=0.5.2), we depend on several packages. To install dependencies using conda:
We use python 3.7. The CUDA version needs to be 10.2. Besides DGL (>=0.8), we depend on several packages. To install dependencies using conda:
```bash
conda create -n Hilander # create env
conda activate Hilander # activate env
conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.2 -c pytorch # install pytorch 1.7 version
conda install -y cudatoolkit=10.2 faiss-gpu=1.6.5 -c pytorch # install faiss gpu version matching cuda 10.2
pip install dgl-cu102 # install dgl for cuda 10.2
pip install dgl-cu102 dglgo -f https://data.dgl.ai/wheels/repo.html # install the latest dgl for cuda 10.2
pip install tqdm # install tqdm
git clone https://github.com/yjxiong/clustering-benchmark.git # install clustering-benchmark for evaluation
cd clustering-benchmark
Expand Down

0 comments on commit 33c8abb

Please sign in to comment.