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 
### 安装 `torch-geometric` 库 为了确保顺利安装 `torch-geometric` 及其相关依赖项,推荐遵循特定的步骤来设置环境并执行安装命令。 #### 创建新的虚拟环境 创建一个新的虚拟环境可以避免与其他项目中的软件包发生冲突,并能更好地控制所使用的 Python 版本以及 PyTorch 的版本。对于大多数情况而言,建议使用 Python 3.6 或更高版本[^3]。 ```bash python -m venv my_pyg_env source my_pyg_env/bin/activate # Linux/macOS my_pyg_env\Scripts\activate # Windows ``` #### 安装 PyTorch 根据目标硬件平台(CPU/GPU),选择合适的 PyTorch 安装方式。这里假设读者已经选择了适合自己的 PyTorch 版本。如果不确定应该选用哪个版本,则可以从官方文档获取最新指导[^1]。 #### 安装 `torch-geometric` 及其依赖项 一旦确认了 PyTorch 已经正确安装完毕,就可以继续安装 `torch-geometric` 和其他必要的扩展库: 针对 CUDA 12.1 和 PyTorch 2.1.0,在 Linux 上可以通过下面这条指令完成安装: ```bash pip install torch_geometric pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.1.0+cu121.html ``` 上述命令会自动处理所有依赖关系,并从指定 URL 下载预编译好的二进制文件以加快安装过程。 另外一种方法是指定 `.whl` 文件直接安装某些组件,例如 `torch_cluster`: ```bash pip install torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl ``` 这种方法适用于那些希望更精确地控制各个子模块具体版本的情况[^2]。 需要注意的是,当尝试通过简单的 `pip install torch-geometric` 来安装时,可能会因为缺少适当构建工具链而导致失败,特别是当本地环境中存在较老版本的 PyTorch 时更容易出现问题。因此强烈建议按照前述指引操作以减少潜在错误的发生几率。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值