图神经网络库(PyTorch geometric:PYG)安装

一、PYG简介

PyTorch Geometric是Pytorch的图深度学习扩展库。

PYG包含已公开发表的论文中关于在图和非规则结构上进行深度学习的各种方法。此外,它还包括一个易于使用的小型批处理加载器、大量通用基准数据集和有用的转换,用于学习任意图以及3D网格或点云。

更多信息,可参考官方文档。

二、使用Anaconda安装pytorch

1、安装Anaconda

在Anaconda官网下载安装包后直接安装即可。

2、安装Pytorch。

在Anaconda Prompt中采用如下命令创建新环境:

conda create -n env_name python=3.7

创建完成后使用如下命令激活新环境:

conda activate env_name

 取pytorch官网查看安装命令

图神经网络库(PyTorch geometric:PYG)安装_第1张图片

此处有两个问题需要注意:

  • 在NVIDIA控制面板查看CUDA版本号,过程如下图所示

图神经网络库(PyTorch geometric:PYG)安装_第2张图片

 

  • 使用清华源加速安装pytorch

 在Anaconda Prompt中输入如下命令:
 

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

 然后打开C:\Users\Administrator 目录下的 .condarc 文件,注释或删除defaults那一行。如下图所示:

图神经网络库(PyTorch geometric:PYG)安装_第3张图片

三、安装PYG

 到pytorch-geometric官网找安装命令,在Anaconda Prompt中输入即可

图神经网络库(PyTorch geometric:PYG)安装_第4张图片

你可能感兴趣的:(图神经网络库(PyTorch geometric:PYG)安装)