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

推荐订阅源

The GitHub Blog
The GitHub Blog
Hugging Face - Blog
Hugging Face - Blog
让小产品的独立变现更简单 - ezindie.com
让小产品的独立变现更简单 - ezindie.com
WordPress大学
WordPress大学
Y
Y Combinator Blog
Google DeepMind News
Google DeepMind News
大猫的无限游戏
大猫的无限游戏
Vercel News
Vercel News
V
Visual Studio Blog
Microsoft Azure Blog
Microsoft Azure Blog
MyScale Blog
MyScale Blog
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
罗磊的独立博客
N
Netflix TechBlog - Medium
M
MIT News - Artificial intelligence
G
Google Developers Blog
阮一峰的网络日志
阮一峰的网络日志
P
Proofpoint News Feed
小众软件
小众软件
Engineering at Meta
Engineering at Meta
D
DataBreaches.Net
S
SegmentFault 最新的问题
H
Help Net Security
量子位

新城旧梦

高中三年复盘 - 新城旧梦 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~~~~~~~~~~~~~~~~~~~~~~~

旧版评论(请先登录):