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版本的成功!

相关推荐
AI科技先锋报5 小时前
2026企业级Data Agent开发平台有哪些:主流DataAgent开发框架全景解析
大数据·人工智能
IT_陈寒6 小时前
Java线程池这破玩意,差点让我周末加班排查到凌晨
前端·人工智能·后端
广州山泉婚姻6 小时前
Go微服务落地:服务通信、注册发现完整实现分享
人工智能·深度学习·微服务
W***25926 小时前
2026 企业 AI 办公工具选型指南:适合团队使用的 AI 办公产品有哪些
大数据·人工智能
码农5996 小时前
让 Agent 替你看 Bug:简单示例读懂 MCP 服务
人工智能
AC赳赳老秦6 小时前
公开音频转写信息提取:OpenClaw 处理发布会与听证会文本并提取核心决策信息
大数据·开发语言·汇编·数据库·人工智能·deepseek·openclaw
四六的六6 小时前
Agent 长会话设计实战:从对话上下文到持久状态,把记忆写进检查清单
人工智能·个人开发·ai编程·ai产品·长上下文·ai代码生成·ai会话
u1301306 小时前
GitHub 热榜项目:周榜(2026-09-27)
人工智能·github
AI搅拌机6 小时前
MiniMax H3导演台Bug修复指南:二采、首尾帧生视频和图生视频优化!全能工作流分享~
人工智能
老纪的技术唠嗑局6 小时前
Tibo 谈 Codex:harness 总比模型快一步
数据库·人工智能