安装torch_geometric

本文档介绍了如何检查CUDA版本和PyTorch版本,并根据官方指南选择合适的PyTorch版本来安装torch_geometric库。对于CUDA 10.1,推荐使用PyTorch 1.7.1。

检查cuda版本

nvidia-smi

 查看pytorch版本

pip list

conda list

版本过低

这里最好去pytorch的官网查看自己需要哪款pytorch

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

参考官网 Installation — pytorch_geometric 2.0.2 documentation

pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${
### 安装 `torch_geometric` 库的方法 要在 AutoDL 环境下安装 `torch_geometric`,需要先确保 PyTorch 和 CUDA 的环境配置正确。以下是详细的说明: #### 配置基础依赖项 为了成功安装 `torch_geometric`,必须首先安装合适的 PyTorch 版本以及对应的 CUDA 工具包。可以参考以下两种方法之一来完成此操作。 1. **指定版本的 PyTorch 安装** 如果目标是安装特定版本的 PyTorch,则可以通过 Conda 命令实现。例如,如果需要安装 PyTorch 1.11.0 并附带相应的工具链,可执行如下命令[^1]: ```bash conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch ``` 2. **动态适配最新版 PyTorch 及其组件** 若希望使用较新的 PyTorch 版本并自动匹配兼容的 cuDNN 组件,可以选择另一种方式[^2]: ```bash conda install pytorch=1.12.0 torchvision=0.13.0 torchaudio=0.12.0 -c pytorch conda install cudnn=6.0 ``` #### 安装 torch_geometric 一旦上述基础依赖已妥善部署完毕,在此基础上继续添加 `torch_geometric` 是可行的。推荐通过 pip 来简化这一过程,具体指令如下所示: ```bash pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-{pytorch_version}+{cuda_version}.html ``` 此处 `{pytorch_version}` 表示所使用的 PyTorch 主次版本号(如 '1.12'),而 `{cuda_version}` 则对应于当前系统的 CUDA 构建标签(比如 'cu113' 对应 CUDA Toolkit 11.3)。对于不支持 GPU 或仅需 CPU 支持的情况,应当替换为 `'cpu'`。 #### 示例脚本 下面提供了一个完整的 Bash 脚本来辅助自动化整个流程: ```bash #!/bin/bash PYTORCH_VERSION="1.12" CUDA_VERSION="cu113" # Step to setup base environment with specified versions of libraries. conda install pytorch=${PYTORCH_VERSION} torchvision torchaudio -c pytorch conda install cudnn=7.6.5 # Proceeding towards installing the geometric package and its dependencies via pip. pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric \ -f https://data.pyg.org/whl/torch-${PYTORCH_VERSION}+${CUDA_VERSION}.html ``` 注意:以上脚本中的变量 `${PYTORCH_VERSION}` 和 `${CUDA_VERSION}` 应依据实际需求调整至恰当数值。 --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值