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

安装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.lryc.cn/news/510459.html

相关文章:

  • Java数组深入解析:定义、操作、常见问题与高频练习
  • Docker-构建自己的Web-Linux系统-镜像webtop:ubuntu-kde
  • 【C语言练习(17)—输出杨辉三角形】
  • SpringMVC学习(二)——RESTful API、拦截器、异常处理、数据类型转换
  • React 第二十节 useRef 用途使用技巧注意事项详解
  • VIVO Java开发面试题及参考答案
  • C# Winfrom chart图 实例练习
  • iOS从Matter的设备认证证书中获取VID和PID
  • 带着国标充电器出国怎么办? 适配器模式(Adapter Pattern)
  • 破解海外业务困局:新加坡服务器托管与跨境组网策略
  • Mybatis-Plus快速入门
  • Chrome被360导航篡改了怎么改回来?
  • Coding(Jenkinsfile)+ Docker 自动化部署 Springboot —— 图文细节和一些注意事项说明
  • docker django uwsgi 报错记录
  • 数据分析思维(五):分析方法——假设检验分析方法
  • 【ES6复习笔记】集合Set(13)
  • 基础爬虫案例实战
  • 深度学习工作:从追求 SoTA 到揭示新现象
  • 学习记录—正则表达式-基本语法
  • 智慧农业物联网传感器:开启农业新时代
  • 普通人怎么入门学习并使用AI?
  • Java中处理if-else的几种高级方法
  • 前端知识补充—CSS
  • 企业架构学习笔记-数字化转型
  • Linux网络——TCP的运用
  • Vue3之状态管理Vuex
  • DPO(Direct Preference Optimization)算法解释:中英双语
  • Hostapd2.11解析笔记
  • js控制文字溢出显示省略号
  • WPF+MVVM案例实战与特效(四十七)-实现一个路径绘图的自定义按钮控件