conda安装gpu版本的pytorch

今天跑代码发现一直是cpu在跑于是检查了一下

因为我是分了区,所以安装代码的依赖的时候没有注意直接安装的cpu版本的torch

于是跑的巨慢,所以找了半天,期间查询了cuda和cudnn,发现都安装上了就差pytorch,遂给出解决方案

1.验证pytorch版本

进入conda后进入你要安装pytorch或者卸载cpu版本pytorch的环境(conda activate xxx)

然后输入如下的内容

如果显示2.x.x而不是2.x.x+cuxxx

那么就是cpu版本的pytorch

2.安装GPU版本pytorch

去pytorch官网

复制代码
https://pytorch.org/get-started/previous-versions/

找到cuda版本的(建议尽量找2.6版本以后的安装,我安装2.5.1的一直给我安装成cpu版本的)

复制命令后注意!!!

要在你安装的环境里复制命令安装

安装的时候会自动帮你卸载原来cpu版本

最后安装完成后继续进行验证,发现安装gpu版本的成功!

相关推荐
J_Xiong01175 小时前
【Agents篇】07:Agent 的行动模块——工具使用与具身执行
人工智能·ai agent
SEO_juper5 小时前
13个不容错过的SEO技巧,让您的网站可见度飙升
人工智能·seo·数字营销
小瑞瑞acd5 小时前
【小瑞瑞精讲】卷积神经网络(CNN):从入门到精通,计算机如何“看”懂世界?
人工智能·python·深度学习·神经网络·机器学习
CoderJia程序员甲6 小时前
GitHub 热榜项目 - 日榜(2026-02-06)
人工智能·ai·大模型·github·ai教程
wukangjupingbb6 小时前
AI多模态技术在创新药研发中的结合路径、机制及挑战
人工智能
CoderIsArt6 小时前
三大主流智能体框架解析
人工智能
民乐团扒谱机6 小时前
【微实验】机器学习之集成学习 GBDT和XGBoost 附 matlab仿真代码 复制即可运行
人工智能·机器学习·matlab·集成学习·xgboost·gbdt·梯度提升树
Coder_Boy_6 小时前
Deeplearning4j+ Spring Boot 电商用户复购预测案例中相关概念
java·人工智能·spring boot·后端·spring
芷栀夏6 小时前
CANN ops-math:揭秘异构计算架构下数学算子的低延迟高吞吐优化逻辑
人工智能·深度学习·神经网络·cann
L543414466 小时前
告别代码堆砌匠厂架构让你的系统吞吐量翻倍提升
大数据·人工智能·架构·自动化·rpa