当前位置:   article > 正文

小白安装pytorch以及pytorch geometirc库以及在Pycharm中使用教程_pytorch自带geometirc

pytorch自带geometirc

pytorch以及pytorch geometric库安装详细教程

在安装的过程中遇到了很多问题,这里总结记录一下,供有相同问题的大家参考解决~

本人安装版本:python3.9.1 ;pytorch1.12.0; pytorch geometric配套pytorch版本

pytorch安装

首先声明pytorch安装的我认为需要提前注意的点:
1.本地安装比官网安装快很多,个人比较推荐本地安装,但是同时需要你注意选择合适的版本
2.如果需要安装pyg库,最好选择1.12.0或者1.11.0(不要选择1.12.1或者1.11.1)

官网安装

本地安装

1.使用anaconda创建环境或者查看现有环境

//创建环境 env_name是环境名 ,x.x是你的python版本
conda create -n env_name python=x.x
//查看现有环境
conda env list
  • 1
  • 2
  • 3
  • 4

2.激活环境

conda activate env_name
  • 1

3.下载要安装的pytorch文件保存在对应的位置
链接: 清华大学开源软件镜像网站pytorch各版本
看好版本,我下载的是cpu版本的,py3.9代表是python3.9对应的
请添加图片描述

4.pip install方式直接安装

pip install 文件名
  • 1

5.验证是否成功

import torch
  • 1

pyg安装

官网安装

参考官方文档输入命令行进行安装
在这里插入图片描述
会出现如下报错,一般是各种关卡限制了网速
在这里插入图片描述
我最后是采用了本地安装的方法成功的~

本地安装

1.下载对应pytorch版本的pyg文件(有四个,分别是cluster、scatter、sparse、spline)
链接: pytorch geometric各版本
文件名后缀cp39代表python3.9版本,win代表电脑为Windows,注意自己下载的是cpu版本还是gpu版本要区分!我这里是cpu版本
请添加图片描述2.pip install方式将这四个文件安装并安装pyg库

 pip install torch_cluster-1.6.0+pt112cpu-cp39-cp39-win_amd64.whl
 pip install torch_sparse-0.6.15+pt112cpu-cp39-cp39-win_amd64.whl
 pip install torch_scatter-2.0.9-cp39-cp39-win_amd64.whl
 pip install torch_spline_conv-1.2.1+pt112cpu-cp39-cp39-win_amd64.whl
 pip install torch-geometric
  • 1
  • 2
  • 3
  • 4
  • 5

如果过程中,速度很慢或者出现这种error可以参考下面的方法进行加速,亲测有效!

参考这篇文章进行加速链接: 加速下载方法
3.验证pyg是否安装成功

import torch
>>> from torch_geometric.data import Data
>>> edge_index =torch.tensor([[0,1,1,2],[1,0,2,1]],dtype=torch.long)
>>> x=torch.tensor([[-1],[0],[1]],dtype=torch.float)
>>> data=Data(x=x,edge_index=edge_index)
>>> print(data)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

输出结果

Data(x=[3, 1], edge_index=[2, 4])
  • 1

在Pycharm中设置及使用pytorch和pyg库

打开pycharm——File(文件)——new project(新建项目)——在项目里创建python file,创建好后,接着,File(文件)——settings(设置)——project(你建的项目名)——project interpreter(python解释器)——add(添加解释器)——添加本地解释器——找到你的anconda里有pytorch和pyg库的环境里的Python
然后在项目里输入

import torch
from torch_geometric.data import Data
  • 1
  • 2

如果没有报错说明可以使用!

以上就是本人从小白安装到最后使用验证的全过程,经历了反反复复的查找资料和思考才算是对整个流程有了较为清晰的认知,所以有时候出现错误反而是促使自己更进一步思考的关键~

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/331608?site
推荐阅读
相关标签
  

闽ICP备14008679号