DGL安装教程

简介: DGL官方安装教程网址:Deep Graph Library以下仅考虑Linux系统的情况。(在Windows上跑GNN是不是太身残志坚了)以下对应可行的安装时间会对应附上。注意:这里对应的cuda版本,一般情况下只要跟cudatoolkit对应上就行。

1. 安装


2022.8.3

我安装PyTorch用的Python3.8,命令是conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=10.2 -c pytorch

可以直接在DGL官网选择对应的情况,复制命令行进行安装:

image.png

注意在能够正常使用之前还需要安装psutil包,否则就会报如下错误:

Traceback (most recent call last):
  File "hello.py", line 1, in <module>
    import dgl.data
  File "my_env/envgraph1/lib/python3.8/site-packages/dgl/__init__.py", line 25, in <module>
    from . import dataloading
  File "my_env/lib/python3.8/site-packages/dgl/dataloading/__init__.py", line 10, in <module>
    from .dataloader import *
  File "my_env/lib/python3.8/site-packages/dgl/dataloading/dataloader.py", line 12, in <module>
    import psutil
ModuleNotFoundError: No module named 'psutil'


安装psutil(psutil的官方下载教程网站:psutil/INSTALL.rst at master · giampaolo/psutil):

pip install psutil

image.png


2. 检验安装效果

import dgl.data
dataset = dgl.data.CoraGraphDataset()
print('Number of categories:', dataset.num_classes)


输出:

my_env/lib/python3.8/site-packages/scipy/__init__.py:146: UserWarning: A NumPy version >=1.16.5 and <1.23.0 is required for this version of SciPy (detected version 1.23.1
  warnings.warn(f"A NumPy version >={np_minversion} and <{np_maxversion}"
Downloading user_path/.dgl/cora_v2.zip from https://data.dgl.ai/dataset/cora_v2.zip...
Extracting file to user_path/.dgl/cora_v2
Finished data loading and preprocessing.
  NumNodes: 2708
  NumEdges: 10556
  NumFeats: 1433
  NumClasses: 7
  NumTrainingSamples: 140
  NumValidationSamples: 500
  NumTestSamples: 1000
Done saving data into cached files.
Number of categories: 7


(这个SciPy和numpy的版本警告我也没辙)

相关文章
|
虚拟化 图形学
A5000 vGPU显示模式切换
A5000显示模式切换
1352 0
A5000 vGPU显示模式切换
|
4月前
|
监控 供应链 数据挖掘
淘宝商品详情API接口解析与 Python 实战指南
淘宝商品详情API接口是淘宝开放平台提供的编程工具,支持开发者获取商品详细信息,包括基础属性、价格、库存、销售策略及卖家信息等。适用于电商数据分析、竞品分析与价格策略优化等场景。接口功能涵盖商品基础信息、详情描述、图片视频资源、SKU属性及评价统计的查询。通过构造请求URL和签名,可便捷调用数据。典型应用场景包括电商比价工具、商品数据分析平台、供应链管理及营销活动监控等,助力高效运营与决策。
306 26
|
6月前
|
算法
一次推理,实现六大3D点云分割任务!华科发布大一统算法UniSeg3D,性能新SOTA
华中科技大学研究团队提出了一种名为UniSeg3D的创新算法,该算法通过一次推理即可完成六大3D点云分割任务(全景、语义、实例、交互式、指代和开放词汇分割),并基于Transformer架构实现任务间知识共享与互惠。实验表明,UniSeg3D在多个基准数据集上超越现有SOTA方法,为3D场景理解提供了全新统一框架。然而,模型较大可能限制实际部署。
408 15
|
11月前
|
机器学习/深度学习 计算机视觉
Siamese网络和Triplet网络
【10月更文挑战第1天】
|
10月前
|
机器学习/深度学习 监控
早停法(Early Stopping)
早停法(Early Stopping)
|
机器学习/深度学习 并行计算 PyTorch
从零开始下载torch+cu(无痛版)
这篇文章提供了一个详细的无痛版教程,指导如何从零开始下载并配置支持CUDA的PyTorch GPU版本,包括查看Cuda版本、在官网检索下载包名、下载指定的torch、torchvision、torchaudio库,并在深度学习环境中安装和测试是否成功。
从零开始下载torch+cu(无痛版)
|
应用服务中间件 Shell 网络安全
nginx安装提示 libssl.so.3: cannot open shared object file: No
【8月更文挑战第1天】### 原因 未将安装的ssl中的`libssl.so.3`链接到`/usr/lib`导致缺失。 ### 解决方案 1. 检查openssl是否已安装,若为低版本则需重装。 ```sh whereis openssl
5462 6
|
机器学习/深度学习 资源调度 自然语言处理
长短时记忆网络(LSTM)完整实战:从理论到PyTorch实战演示
长短时记忆网络(LSTM)完整实战:从理论到PyTorch实战演示
16331 0
|
存储 人工智能 自然语言处理
【AI大模型】Transformers大模型库(五):AutoModel、Model Head及查看模型结构
【AI大模型】Transformers大模型库(五):AutoModel、Model Head及查看模型结构
640 0
|
Python
【Python】已完美解决:(Python键盘中断报错问题) KeyboardInterrupt
【Python】已完美解决:(Python键盘中断报错问题) KeyboardInterrupt
781 3

热门文章

最新文章