【Lunix】pytorch-geometric GPU 版安装

最新推荐文章于 2024-02-08 20:01:28 发布

妈妈这道题好难

最新推荐文章于 2024-02-08 20:01:28 发布

阅读量229

点赞数

分类专栏：图神经网络环境安装 bug处理文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/sinat_25479505/article/details/126176910

版权

前言

随着图神经网络兴起，一些类似dgl图神经训练工具越发重要，本文主要介绍基于pytorch的图神经网络框架-GPU版的安装

环境

Lunix
Anaconda

安装

笔者根据官网进行安装，很容易出现各种问题，读者可以尝试通过yaml文件进行，gnn环境的一次构建

conda gnn 虚拟环境构建

# gnn.yaml文件

name: gnn
channels:
  - conda-forge
dependencies:
  - python=3.9.10
  - pip=22.0.3
  - setuptools=59.5.0
  - click
  - cudatoolkit=11.3.1
  - cupy
  - rapidsai::cudf
  - pytorch::pytorch=1.10.2
  - torchmetrics  # modular ML metrics for PyTorch
  - pytorch-lightning  # parallelisation
  - pyg::pyg=2.0.3  # pytorch geometric
  - tensorboard  # visualising metrics
  - numpy=1.20
  - pandas<1.4.0
  - omegaconf
  - tqdm
  - "hdf5=1.*=mpi_openmpi*"
  - "h5py=3.*=mpi_openmpi*"
  - ipython=7.28.0
  - jupyter
  - pythia8
  - attrs
  - networkit
  - p