Bootstrap

torch-geometric/dgl安装

安装pytorch

根据py版本,下载whl

对应的 torch;torchaudio;torchvision 版本查看网址:https://pytorch.org/get-started/previous-versions/

(注意:gpu版的需要提前下载cuda)

我的cuda是11.2可以用11.1的

下载网站如下

download.pytorch.org/whl/cu113

下载好了如下

在anaconda中进入虚拟环境

安装命令,由于我的轮子下载在D:\whl\文件夹下,安装

pip install D:\whl\torch-1.9.1+cu111-cp39-cp39-win_amd64.whl
pip install D:\whl\torchaudio-0.9.1-cp39-cp39-win_amd64.whl
pip install D:\whl\torchvision-0.10.1+cu111-cp39-cp39-win_amd64.whl

可进入python环境测试是否安装成功

import torch

无报错即安装成功

安装torch-geometric

根据torch版本安装轮子,网址data.pyg.org/whl/

同样pip命令安装这些whl

安装这四个whl后安装pyg包

pip install torch_geometric

报错Read timed out.运行下面代码控制超时即可

pip --default-timeout=100 install torch_geometric

测试是否安装成功

import torch_geometric

无报错即安装成功

报错AttributeError: 'builtin_function_or_method' object has no attribute 'default'

原因:

torch_geometric 2.6.1 是一个相对较新的版本,而我的 PyTorch 版本是 1.9.1

卸载torch-geometric

pip uninstall torch-geometric

再次安装

pip install torch-geometric==2.0.4

报错× python setup.py egg_info did not run successfully.

降级 setuptools

pip install setuptools==59.5.0

后再次安装

pip install torch-geometric==2.0.4

安装成功

安装dgl

同样步骤,安装pytorch

安装torchdata;torch和torchdata对应版本如下 torchdata · PyPI

# 查看可以使用的dgl版本

conda search dgl -c dglteam

安装

conda install -c dglteam dgl=0.6.0

检查是否有报错

numpy版本Version: 1.23.4

;