栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【Lunix】pytorch-geometric CPU 版安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【Lunix】pytorch-geometric CPU 版安装

前言

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

环境
  • Lunix
  • Anaconda
安装
  • python3.8 基础环境安装
conda create -n GNN_envir python=3.8

source avtivate GNN_envir
  • pytorch-geometric 依赖环境安装

         建议安装最新版的,低版本的安装容易受到文档、包间版本严格依赖等限制导致安装困难。虽然网上有各种解决办法,但是不同机器环境不同,每个人操作复现起来不一定能成功,所以这里直接用的官网最新版本进行安装

 这里pip版本需要和python版本相对应,若全局找不到pip,一版存在anconda根目录下envs/GNN_envir/bin/pip

pip3.8 install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.9.0+cpu.html
  • pytorch-1.9安装

第二部依赖包安装是不会安装对应版本PyTorch,并且和网上大家所查到的坑就是pip install自动安装的PyTorch不主动区分cpu和gpu版本,会导致安装后的程序无法使用,所以本文选择在对应网站下载所需环境对应的PyTorch版本whl

wget https://download.pytorch.org/whl/cpu/torch-1.9.1%2Bcpu-cp38-cp38-linux_x86_64.whl

pip3.8 install torch-1.9.1+cpu-cp38-cp38-linux_x86_64.whl
  •  环境测试

  •  Jupyter加入环境

需要切换到已创建的conda环境

pip3.8 install  ipykernel

python3.8 -m ipykernel install --user --name GNN_envir --display-name GNN_envir
FAQ 

一般情况下,上述能够正常安装后使用,但运行时难免会遇到一些问题,如

/lib64/libstdc++.so.6: version `GLIBCXX_3.4.21' not found

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/674595.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号