如何安装cuda版本的torch-sparse和torch-scatter

安装对应cuda版本的torch,确保cuda可用

使用nvidia-smi查看cuda版本,我的是11.4,然后就找到pytorch历史版本,页面搜索cuda 11.4,没搜到,继续往小版本搜,搜到cuda 11.3,果断安装:
访问这里找到对应你cuda版本的torch
在这里插入图片描述

找到对应torch+cuda版本的torch-scatter和torch-sparse

因为我的版本是1.12.1+cu113,所以,按照此页面的指引,运行以下命令即可。如果你要安装自己的版本,可以替换掉我下面的1.12.1+cu113为自己的版本。

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.12.1+cu113.html
### 官方下载链接 对于希望获取 `torch-geometric`、`torch-scatter` 及 `torch-sparse` 的官方版本,推荐的方式是从 PyTorch Geometric 提供的仓库中找到适合特定 CUDA 版本以及 Python 解释器版本的文件。通常情况下,这些库依赖于特定版本PyTorch 并且可能需要匹配相同的 CUDA 工具包版本。 #### 获取适用于指定配置的轮子(Whl) 为了确保兼容性,建议先确认已安装PyTorch 版本及其所基于的 CUDA 版本: ```bash python -c "import torch; print(torch.__version__, torch.version.cuda)" ``` 接着访问 [PyTorch Geometric 轮子页面](https://data.pyg.org/whl/) 或者更稳定的长期支持版面如 [PyTorch Stable Releases](https://pytorch-geometric.com/whl/torch_stable.html),从中挑选与当前环境中 PyTorch CUDA 版本相匹配的 `.whl` 文件进行下载安装[^1]。 例如,如果使用的是带有 CUDA 11.3 支持的 PyTorch 1.10.1,在 Linux 上运行 Python 3.7,则可以通过如下命令来安装所需的组件: ```bash pip install https://data.pyg.org/whl/torch-1.10.1+cu113/torch_cluster-1.5.8-cp37-cp37m-linux_x86_64.whl pip install https://data.pyg.org/whl/torch-1.10.1+cu113/torch_scatter-2.0.9-cp37-cp37m-linux_x86_64.whl pip install https://data.pyg.org/whl/torch-1.10.1+cu113/torch_sparse-0.6.12-cp37-cp37m-linux_x86_64.whl pip install torch_geometric ``` 请注意 URL 中的具体版本号应当依据实际需求调整,并且确保所有软件包之间的版本相互兼容[^4]。 最后一步直接通过 `pip install torch_geometric` 来完成对 `torch-geometric` 自身的安装,这会自动处理其最低限度的需求项。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值