【python+pycharm+pip安装torch_geometric】

安装torch_geometric: python+pycharm+pip

本文针对python(无anaconda)+pip方式安装cpu环境下的torch_geometric
pyg的库:添加链接描述

第一步:查看自己的torch版本(以torch1.4.0为例)
第二步:
准备torch-sparse、torch-scatter、torch-cluster、 torch-spline-conv whl文件
进入网址:https://pytorch-geometric.com/whl/torch-1.4.0+cpu.html
(不同的torch版本改一下就可以了)
找到自己要下载的版本(注意要和python版本匹配,如python3.7对应cp37),直接下载whl文件(默认存储在C盘User目录下,放在别的位置好像会出问题,总之别改就对了)
在这里插入图片描述
第三步:pip安装
依次输入四句命令:
pip install torch_scatter-2.0.4-cp37-cp37m-win_amd64.whl
pip install torch_sparse-0.6.0-cp37-cp37m-win_amd64.whl
pip install torch_cluster-1.5.3-cp37-cp37m-win_amd64.whl
pip install torch_spline_conv-1.2.0-cp37-cp37m-win_amd64.whl
在这里插入图片描述
这四个都成功安装之后,进入最后一步
pip install torch-geometric==1.5.0(第一次这里没有指定版本,结果安装了一个2.2.0的版本,最后在pycharm里面检验的时候出现了问题,后来指定了1.5.0之后就好了)
在这里插入图片描述
至此,安装成功!(但还不算真正的成功)在这里插入图片描述
第四步:cmd里安装成功后,进入pycharm检验一下
在这里插入图片描述
如果有问题的话,大概率是torch、torch_geometric、torch_sparse、torch_scatter、torch_cluster、 torch_spline_conv版本不匹配,所以正式安装之前,一定要弄清楚自己要的版本以及这些库之间是否相互匹配的问题。

在安装过程中踩了很多雷,看了很多经验贴之后的一个总结,仅供参考。对于版本问题,因为我是跑文献里面的代码,作者是给出了所有库的版本要求的,所以直接下载对应的版本就好了。

  • 4
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装GPU版本的Torch可以通过以下步骤进行: 1. 首先,确保你的计算机上已经安装了NVIDIA显卡驱动程序,并且支持CUDA。你可以在NVIDIA官方网站上下载并安装最新的显卡驱动程序。 2. 接下来,安装CUDA工具包。你可以在NVIDIA官方网站上下载适合你显卡的CUDA版本,并按照官方文档的指导进行安装。 3. 安装cuDNN库。cuDNN是一个针对深度神经网络的GPU加速库,可以提高Torch在GPU上的性能。你可以在NVIDIA官方网站上下载适合你CUDA版本的cuDNN库,并按照官方文档的指导进行安装。 4. 确保你的Python环境已经安装pip包管理器。然后使用以下命令安装GPU版本的Torchtorchvision: ``` pip install torch torchvision ``` 5. 安装完成后,你可以使用以下命令验证Torch是否成功安装并可用: ``` python import torch print(torch.cuda.is_available()) ``` 如果输出为True,则表示Torch已经成功安装并可以在GPU上运行了。 请注意,以上步骤是基于使用pip包管理器进行安装的。如果你使用的是conda包管理器,可以使用类似的命令进行安装,但可能会有一些细微的差异。 #### 引用[.reference_title] - *1* *2* *3* [Cuda 和 GPU版torch安装最全攻略,以及在GPU 上运行 torch代码](https://blog.csdn.net/weixin_46833029/article/details/127946022)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值