Windows10安装torch-geometric

查看 pytorch, cuda, cudnn的版本

在命令行中输入python转入python命令行
按照图中分别输入,查看pytorch, cuda, cudnn的版本

在这里插入图片描述
网站点击对应的条目
例如pytorch版本为1.4, cuda版本为102,因此选择torch-1.5.0+cu102

在这里插入图片描述
在之后跳出来的网站里分别点击下载四个文件:scatter / sparse / cluster / spline-conv

下载好后,记住文件存放的地址,分别进行安装
例如,上述文件存放地址为:D:\python\
则依次在命令行中输入

pip install D:\python\torch_spline_conv-1.2.0-cp37-cp37m-win_amd64.whl
pip install D:\python\torch_cluster-1.5.4-cp37-cp37m-win_amd64.whl
pip install D:\python\torch_scatter-2.0.4-cp37-cp37m-win_amd64.whl
pip 
为了安装与当前 PyTorch 1.11.0 版本兼容的 `torch-geometric` 及其相关依赖(如 `torch-sparse`, `torch-cluster`, `torch-scatter` 等),需要手动查找与 PyTorch 1.11.0 兼容的对应版本。官方安装页面通常只提供最新版本的安装命令,因此需要通过以下方法确定合适的版本组合。 ### 版本匹配策略 1. **查看 PyTorch 官方版本发布时间**: PyTorch 1.11.0 发布于 202110 月左右,因此应寻找在该时间段前后发布的 `torch-geometric` 及其依赖库版本。 2. **访问 PyTorch Geometric 的 GitHub 发布页面**: - 前往 [PyTorch Geometric 的 GitHub Releases 页面](https://github.com/pyg-team/pytorch_geometric/releases) - 查找与 PyTorch 1.11.0 对应的 `torch-geometric` 版本,通常为 `1.8.x` 或 `1.7.x`。 3. **查找对应的安装命令**: 根据经验,以下版本组合通常兼容 PyTorch 1.11.0- `torch-sparse`: 0.6.12 - `torch-cluster`: 1.5.9 - `torch-scatter`: 2.0.9 - `torch-geometric`: 1.8.0 安装命令如下(以 CUDA 11.3 为例): ```bash pip install torch-sparse==0.6.12+cu113 -f https://data.pyg.org/whl/torch-1.11.0.html pip install torch-cluster==1.5.9+cu113 -f https://data.pyg.org/whl/torch-1.11.0.html pip install torch-scatter==2.0.9+cu113 -f https://data.pyg.org/whl/torch-1.11.0.html pip install torch-geometric==1.8.0 ``` 4. **验证安装**: 安装完成后,可通过以下代码验证是否成功导入: ```python import torch import torch_geometric print(torch.__version__) print(torch_geometric.__version__) ``` 5. **处理 DLL 初始化失败问题**: 如果出现 `OSError: [WinError 1114] 动态链接库(DLL)初始化例程失败`,通常是由于 PyTorch 与扩展库的 CUDA 或构建版本不一致。建议: - 确保所有包均使用相同的 CUDA 和 PyTorch 构建版本。 - 使用官方推荐的安装方式,避免混用不同来源的 wheel 文件。 -Windows 上,考虑使用 Conda 环境来管理依赖以避免 DLL 冲突。 ### 总结 安装与 PyTorch 1.11.0 兼容的 `torch-geometric` 需要手动查找对应版本并使用指定安装命令。推荐使用上述版本组合,并确保所有依赖项版本一致以避免兼容性问题。 ---
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值