惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

Recent Announcements
Recent Announcements
V
V2EX
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
博客园 - 聂微东
爱范儿
爱范儿
Jina AI
Jina AI
博客园 - Franky
IT之家
IT之家
奇客Solidot–传递最新科技情报
奇客Solidot–传递最新科技情报
T
Tailwind CSS Blog
博客园 - 三生石上(FineUI控件)
The Cloudflare Blog
M
MIT News - Artificial intelligence
aimingoo的专栏
aimingoo的专栏
让小产品的独立变现更简单 - ezindie.com
让小产品的独立变现更简单 - ezindie.com
J
Java Code Geeks
人人都是产品经理
人人都是产品经理
腾讯CDC
博客园_首页
月光博客
月光博客
有赞技术团队
有赞技术团队
C
Check Point Blog
Microsoft Security Blog
Microsoft Security Blog
MyScale Blog
MyScale Blog

新城旧梦

高中三年复盘 - 新城旧梦 2024年终总结 - 新城旧梦 高数上笔记 - 新城旧梦 鸿蒙开发记录 - 新城旧梦 git仓库迁移并保留所有记录 - 新城旧梦 记一次ssh无法登陆的处理 - 新城旧梦 amftp 3.0 for docker 发布 - 新城旧梦 Amysql 1.7 for Docker发布 - 新城旧梦 2023新年快乐 - 新城旧梦
Anaconda环境下安装PYG及CUDA - 新城旧梦
dadafox · 2025-06-17 · via 新城旧梦

一、Pytorch主体安装

  1. 以管理员身份打开Anaconda Prompt

  2. 输入conda install pytorch::pytorch
    conda install pytorch::pytorch
    若中间有任何提示全部输入y

  3. 验证安装
    输入python进入python环境
    输入如下以导入pytorch并输出版本

    import torch
    torch.__version__

    若有输出则安装正确,如下图

请确保电脑为英伟达GPU并且已经安装显卡驱动

  1. 查看驱动版本
    打开cmd,输入nvidia-smi
    查看输出的CUDA版本

    这里输出的是驱动对应的最高版本,我们所安装的CUDA版本需要小于等于它,建议和输出的版本一致

  2. 下载CUDA
    点击前往CUDA官网
    选择对应的版本(以12.8.0为例)

    选择对应的系统版本,安装方式建议选择local

  3. 安装CUDA
    选择自定义

    全部勾选,并下一步直至安装完成

  4. 验证安装
    cmd输入nvcc -V若有输出则安装完成

三、PYG组件安装

点击访问下载地址

  1. 下载5个PYG小组件
    首先根据自己的pytorch版本号和cuda版本号进入相对应的文件夹

    必须要确认pytorch和cuda的版本号一致

进入文件夹后选择自己系统版本和python版本对应的最高版本下载

Anaconda截止目前需下载cp12的版本

  1. 安装组件
    依次安装一下5个组件
    pyg_lib,torch_cluster,torch_scatter,torch_sparse,torch_spline_conv
    首先,以管理员身份运行anaconda命令行(参见文章顶部图1),
    依次输入pip install xxxxx(下载的具体文件名)
    进行安装即可

四、安装PYG本体

在anaconda命令行输入
pip install torch_geometric
即可完成安装

分割线QAQ~~~~~~~~~~~~~~~~~~~~~~~

旧版评论(请先登录):