本文介绍了PyTorch Geometric（PYG）的基本概念，它是一个PyTorch的图深度学习扩展库，提供了多种图学习方法。接着详细讲解了如何使用Anaconda来安装PyTorch和PYG，包括Anaconda的安装、PyTorch环境的创建与激活，以及根据CUDA版本和使用清华源加速安装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官网查看安装命令
此处有两个问题需要注意：
- 在NVIDIA控制面板查看CUDA版本号，过程如下图所示
最低0.47元/天 解锁文章
扫一扫
621
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
