Anaconda环境下安装PYG及CUDA

一、Pytorch主体安装

1. 以管理员身份打开Anaconda Prompt
   

2. 输入conda install pytorch::pytorch
   conda install pytorch::pytorch
   若中间有任何提示全部输入y
   

3. 验证安装
   输入python进入python环境
   输入如下以导入pytorch并输出版本

   import torch
   torch.__version__

   若有输出则安装正确，如下图
   

二、CUDA安装

请确保电脑为英伟达GPU并且已经安装显卡驱动

1. 查看驱动版本
   打开cmd，输入nvidia-smi
   查看输出的CUDA版本
   
   这里输出的是驱动对应的最高版本，我们所安装的CUDA版本需要小于等于它，建议和输出的版本一致

2. 下载CUDA
   点击前往CUDA官网
   选择对应的版本（以12.8.0为例）
   
   选择对应的系统版本，安装方式建议选择local
   

3. 安装CUDA
   选择自定义
   
   全部勾选，并下一步直至安装完成

4. 验证安装
   cmd输入nvcc -V若有输出则安装完成
   

三、PYG组件安装

点击访问下载地址

1. 下载5个PYG小组件
   首先根据自己的pytorch版本号和cuda版本号进入相对应的文件夹

   必须要确认pytorch和cuda的版本号一致

进入文件夹后选择自己系统版本和python版本对应的最高版本下载

Anaconda截止目前需下载cp12的版本

2. 安装组件
   依次安装一下5个组件
   pyg_lib，torch_cluster，torch_scatter，torch_sparse，torch_spline_conv
   首先，以管理员身份运行anaconda命令行（参见文章顶部图1），
   依次输入pip install xxxxx（下载的具体文件名）
   进行安装即可

四、安装PYG本体

在anaconda命令行输入
pip install torch_geometric
即可完成安装