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的版本警告我也没辙)

相关文章
|
6月前
|
PyTorch 算法框架/工具 Python
pytorch安装教程
pytorch安装教程
139 0
|
并行计算 PyTorch 算法框架/工具
Python安装Pytorch教程(图文详解)
Python安装Pytorch教程(图文详解)
1892 0
Python安装Pytorch教程(图文详解)
|
并行计算 异构计算
下载和安装CUDA和Cudnn(图文详解)
下载和安装CUDA和Cudnn(图文详解)
1234 0
下载和安装CUDA和Cudnn(图文详解)
|
机器学习/深度学习 并行计算 PyTorch
PyTorch安装教程
PyTorch是学习深度学习时常用的Python神经网络框架,本文将介绍其部分版本的安装方式。Windows和Linux通用。 作者使用anaconda作为管理虚拟环境的工具。以下工作都在虚拟环境中进行,对Python和Aanaconda的安装及对虚拟环境的管理本文不作赘述,后期可能会撰写相关的博文。
PyTorch安装教程
|
1月前
|
PyTorch TensorFlow 算法框架/工具
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。
51 1
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
|
5月前
|
TensorFlow 算法框架/工具 Python
最新版tensorflow安装教程,pip安装+手动安装
最新版tensorflow安装教程,pip安装+手动安装
161 1
|
6月前
|
并行计算 PyTorch 算法框架/工具
pytorch的安装步骤
pytorch的安装步骤
172 6
|
机器学习/深度学习 数据可视化 数据挖掘
PyTorch Geometric (PyG) 入门教程
PyTorch Geometric是PyTorch1的几何图形学深度学习扩展库。本文旨在通过介绍PyTorch Geometric(PyG)中常用的方法等内容,为新手提供一个PyG的入门教程。
PyTorch Geometric (PyG) 入门教程
|
自然语言处理 并行计算 PyTorch
基于Pytorch中安装torchvision简单详细完整版
基于Pytorch中安装torchvision简单详细完整版
2624 1
基于Pytorch中安装torchvision简单详细完整版
|
存储 并行计算 PyTorch
基于Pytorch中安装torch_geometric简单详细完整版
基于Pytorch中安装torch_geometric简单详细完整版
1303 0
基于Pytorch中安装torch_geometric简单详细完整版
下一篇
无影云桌面