安装与cuda版本对应的torch-geometric文件的shell

最新推荐文章于 2025-04-23 17:31:19 发布

代码搬运工易辰

最新推荐文章于 2025-04-23 17:31:19 发布

阅读量235

点赞数 8

CC 4.0 BY-SA版权

分类专栏： #+代码漏洞检测文章标签： pytorch 人工智能 python

本文链接：https://blog.youkuaiyun.com/wangyifan4576/article/details/141614305

#+代码漏洞检测专栏收录该内容

6 篇文章

订阅专栏

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')