nvcc.conda activate base
mkdir pygeo
cd pygeo
git clone --recursive https://github.com/rusty1s/pytorch_geometric.git
git clone --recursive https://github.com/rusty1s/pytorch_cluster.git
git clone --recursive https://github.com/rusty1s/pytorch_scatter.git
git clone --recursive https://github.com/rusty1s/pytorch_spline_conv.git
git clone --recursive https://github.com/rusty1s/pytorch_sparse.git
python ./pytorch_sparse/setup.py develop
python ./pytorch_spline_conv/setup.py develop
python ./pytorch_scatter/setup.py develop
python ./pytorch_cluster/setup.py develop
python ./pytorch_geometric/setup.py develop
I believe there are is also a dependency called METIS, but it is optional. If you need to satisfy conda dependencies, I usually just write a stub meta.yaml in an empty folder with build.sh also doing nothing.
Also to note, due to CUDA you need GCC 8.4 or below. I've tested it to be working with torch 1.4.0 and torch 1.5.0dev (latest master branch)
You can also just use the provided wheels for an easier installation.
Thank you for your help!
Most helpful comment
I believe there are is also a dependency called METIS, but it is optional. If you need to satisfy conda dependencies, I usually just write a stub meta.yaml in an empty folder with build.sh also doing nothing.
Also to note, due to CUDA you need GCC 8.4 or below. I've tested it to be working with torch 1.4.0 and torch 1.5.0dev (latest master branch)