I am doing my independent study about clustering by GNN method. Here is github link of the paper. I want to use GPU to run all .ipynb files in jupyter notebook. I use Main.ipynb as example. My computer environment is