You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

无法下载使用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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.22 22:54:17