安装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