import logging
import subprocess
from setuptools import setup
import torch
cuda_v = f"cu{torch.version.cuda.replace('.', '')}"
torch_v = torch.__version__.split('.')
torch_v = '.'.join(torch_v[:-1] + ['0'])
def system(command: str):
output = subprocess.check_output(command, shell=True)
logging.info(output)
system(f'pip install scipy')
system(f'pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-{torch_v}+{cuda_v}.html')
system(f'pip install torch-geometric')
05-10
1640
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
12-05
1952
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
01-29
2977
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
02-14
1312
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)