Pytorch实战---NIMCGCN

---

---

1. 问题背景

1. 已在Anaconda中安装好Pytorch1.7.1cpu版本
2. 运行NIMCGCN代码

---

2. 安装torch_geometric

• windows安装torch_geometric教程
  依次使用pip安装

pip install isodate
pip install rdflib
pip install plyfile
pip install pytest-runner
pip install torch_geometric

运行发现显示缺少torch_sparse和torch_scatter

• 该问题参考博客
  点击此处下载scatter和sparse文件
  

  • 根据Pytorch版本选择对应版本，cpu版本选择后缀cpu，gpu版本选择后缀cu+版本(CUDA版本)
    

  • 根据Pytorch所在环境的python版本和系统选择对应cp+版本

• 下载对应文件存放于C盘user\用户名目录下，使用pip安装对应文件

  pip install torch_sparse-0.6.8-cp37-cp37m-win_amd64.whl
  pip install torch_scatter-2.0.5-cp37-cp37m-win_amd64.whl
  

• 完成后即可正确运行

---

3. 适配CPU版本Pytorch

• 修改数据集路径和结果保存路径
  数据集路径修改
  结果保存路径
  
• CPU版本和GPU版本运行问题及其修改
  由于笔者电脑没有独立显卡，安装的CPU版本的Pytorch，而代码是GPU版本的代码，将代码中所有的.CUDA()修改为.cpu()再运行。
  

---