图神经网络(GNN)

安装

  1. 创建一个python虚拟环境:可用virtualenv或者miniconda创建,这里我安装3.11
  1. 安装pytorch
  • 官方包的github地址:pytorch包的github地址
  • 选择合适的配置安装命令:地址
    在这里插入图片描述
  • 查看torch的版本:终端运行python -c "import torch; print(torch.__version__)"
  • 查看CUDA的版本:终端运行python -c "import torch; print(torch.version.cuda)
  1. 安装torch_geometric及相关wheels(参考链接:官方文档
  • 先安装torch_geometric:pip install torch_geometric
  • 选择合适的配置,然后安装wheels/dependencies
    在这里插入图片描述
Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐