图神经网络开发环境配置记录

最新推荐文章于 2023-02-06 11:29:22 发布

格维小康

最新推荐文章于 2023-02-06 11:29:22 发布

阅读量409

点赞数

分类专栏：个人学习记录文章标签： pytorch python 神经网络机器学习

本文链接：https://blog.csdn.net/Gewe_Xiaokang/article/details/114947135

版权

本文是一位大四学生配置图神经网络开发环境的过程记录，包括安装PyTorch及其特定版本、解决pytorch-geometric安装错误以及在IDE中调用torch的问题。遇到的问题包括找不到对应CUDA版本的PyTorch、pytorch-geometric安装报错和IDE无法识别已安装的torch。通过查阅资料和调整安装步骤，最终成功配置完成。

摘要由CSDN通过智能技术生成

做毕设的大四生一枚。本文主要记录在配置中出现的问题及解决方式，逻辑结构可能混乱，作为答疑帖有些不合格，但如果能帮到别人，那就太好了。

配置主要参考：
①《从代码的角度深入浅出图神经网络（GNN）第二期开发环境&图结构数据初始化》——B站up主：I思凡I
https://www.bilibili.com/video/BV1zt4y1U7fM/
②从代码角度深入浅出图神经网络系列笔记（一）——CSDN作者：Sp4rkW https://blog.csdn.net/wy_97/article/details/108489943

问题1：无法直接在cmd中安装对应版本。
我安装的是cu110,torch1.7.0版本的，如果直接按照教程安装pytorch的话，通过pip install指令会显示找不到对应cu110，torch1.7.0版本。

解决方法：参考了 https://blog.csdn.net/ccutyear/article/details/106507310 的方法，根据提示到 https://download.pytorch.org/whl/torch_stable.html 网站找对应的版本，并手动安装即可，即不在cmd窗口在线下载。

问题2：pytorch-geometric安装出现错误。
其实torch安装不难，关键是后续要兼容pytorch-geometric的安装。因为不能安装这个库，我重新安装了好几遍torch。
常见问题即在执行这些指令的时候报错，每一条都有可能出错。