当前位置: 首页 > article >正文

安装torch-geometric库

目录

1.查看 torch 和 CUDA 版本

2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv

3.下载并安装torch-geometric库


1.查看 torch 和 CUDA 版本

查看CUDA版本

nvcc -V

查看pytorch版本

pip show torch

2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv

 网址: https://data.pyg.org/whl/

 根据自己的torch版本和CUDA选择

选择需要的版本,右键复制链接。

在命令行使用 wget 下载链接

wget -P 目标地址 https://data.pyg.org/whl/torch-2.0.0%2Bcu118/torch_sparse-0.6.18%2Bpt20cu118-cp310-cp310-linux_x86_64.whl

下载以上五个库的wheel文件之后,使用 pip install 安装

pip install wheel文件地址

3.下载并安装torch-geometric库

网址:PyG

 

根据自己的torch和cuda版本安装 torch-geometric库

pip install torch_geometric -f https://data.pyg.org/whl/torch-{torch版本号}+cu{CUDA版本号}.html
# pip install torch_geometric -f https://data.pyg.org/whl/torch-2.0.0+cu118.html

安装完成,无报错


http://www.kler.cn/a/459910.html

相关文章:

  • 后端开发-Maven
  • 第2章:SQL基础
  • 计算机网络•自顶向下方法:网络层介绍、路由器的组成
  • SpringCloudAlibaba实战入门之路由网关Gateway过滤器(十三)
  • RAG实战:本地部署ragflow+ollama(linux)
  • Codigger集成Copilot:智能编程助手
  • 正则表达式:高级应用与性能优化
  • uniapp使用ucharts组件
  • 21天掌握JavaWeb - 第17天:前端页面开发与集成测试
  • leetcode 热题100(78. 子集)dfs回溯 c++
  • #渗透测试#红蓝攻防#红队打点web服务突破口总结02
  • HTML——23. 锚点和空链接二
  • 单片机理论基础
  • InstructGPT:基于人类反馈训练语言模型遵从指令的能力
  • Hadoop HA安装配置(容器环境),大数据职业技能竞赛模块A平台搭建,jdk+zookeeper+hadoop HA
  • 牛津Meta最新!PartGen:基于多视图扩散模型的多模态部件级3D生成和重建!
  • 网络安全行业研究报告
  • XDOJ 767 哈弗曼树
  • VBA批量插入图片到PPT,一页一图
  • 【uniapp】APP内嵌webview消息传递
  • 计算机网络 (13)信道复用技术
  • SmartAIChain荣获重要认可
  • MATLAB符号计算-符号表达式基础运算操作
  • CAD学习 day11
  • 【数据结构】线性数据结构——链表
  • 深入了解 Linux tree 命令及其常用选项:Linux如何显示目录结构和文件大小