无法下载使用OGB且代码持续运行无输出的问题求助
解决OGB数据集导入/下载卡住无输出的问题
针对你在Windows11+PyCharm环境下,执行OGB相关代码时卡住无输出的问题,提供以下几种可行的解决办法:
1. 手动下载数据集并指定本地路径
Windows环境下自动下载可能因网络或权限问题卡住,可手动完成数据集部署:
- 下载
ogbn-arxiv数据集文件,放到本地目录(默认路径为C:\Users\你的用户名\.ogb\nodeproppred\ogbn_arxiv,若目录不存在需手动创建) - 修改代码,通过
root参数指定本地数据集路径:
import torch_geometric.transforms as T from ogb.nodeproppred import PygNodePropPredDataset dataset_name = 'ogbn-arxiv' # 替换为你的本地数据集路径 dataset = PygNodePropPredDataset(name=dataset_name, transform=T.ToSparseTensor(), root='C:/Users/你的用户名/.ogb/nodeproppred/') print('The {} dataset has {} graph'.format(dataset_name, len(dataset))) data = dataset[0] print(data)
2. 排查PyCharm代理设置
若PyCharm配置了无效代理,会导致数据集下载请求受阻:
- 打开
File > Settings > Appearance & Behavior > System Settings > HTTP Proxy - 切换为
No proxy模式,再重新运行代码测试
3. 以管理员身份运行PyCharm
Windows的权限限制可能导致OGB无法创建或写入数据集缓存目录:
- 右键点击PyCharm图标,选择「以管理员身份运行」,再执行代码
4. 验证依赖版本兼容性
即使满足官网基础要求,版本不匹配也可能引发异常:
- 执行以下命令查看当前依赖版本:
pip list | findstr "torch-geometric ogb torch"
- 确保
torch-geometric与ogb版本兼容(例如ogb==1.3.5对应torch-geometric>=2.0.0),版本不符时重新安装:
pip install ogb==1.3.5 torch-geometric==2.3.0
5. 配置OGB环境变量规避缓存问题
通过设置环境变量,强制指定缓存目录或禁用自动下载验证:
import os # 指定缓存目录,替换为你的路径 os.environ['OGB_CACHE_DIR'] = 'C:/Users/你的用户名/.ogb/' # 手动下载完成后可设为1禁用自动下载 os.environ['OGB_DISABLE_DOWNLOAD'] = '0' import torch_geometric.transforms as T from ogb.nodeproppred import PygNodePropPredDataset dataset_name = 'ogbn-arxiv' dataset = PygNodePropPredDataset(name=dataset_name, transform=T.ToSparseTensor()) print('The {} dataset has {} graph'.format(dataset_name, len(dataset))) data = dataset[0] print(data)
内容的提问来源于stack exchange,提问作者ChenTingyang
相关产品推荐
相关产品推荐

