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

相关推荐
fthux5 小时前
装闭 RenoPit 源码解析(07):装修闭坑知识库与AI Prompt构建
人工智能·ai·开源·github·open source·renopit
zyplayer-doc7 小时前
VuePress类静态文档站和动态知识库怎么选:两种技术路线的适用场景
javascript·人工智能·后端·安全·智能手机
KKKlucifer8 小时前
拨开接口黑盒迷雾:运营商第三方合作接口安全审计与准入管控落地实践
网络·人工智能·安全
梦梦代码精8 小时前
连锁品牌数字化:从门店扩张到用户资产运营的技术底座
大数据·人工智能·低代码·docker·开源·代码规范
咕噜咕噜啦啦8 小时前
vLLM框架
人工智能·qwen·vllm
启雀AI8 小时前
AI 驱动的视频课程自动摘要与知识点提取:ASR + LLM 流水线工程实践
人工智能·阿里云·华为云·音视频·培训saas平台
Mac的实验室8 小时前
2026年8月最新实操:谷歌Gmail邮箱手机号注册扫码发短信提示“无法验证”怎么办?(附100%成功绕过指南)
人工智能
MindUp9 小时前
AI辅助PPT生成工具的内容组织能力实测:8款产品的文档解析与排版效果对比
人工智能
寒草9 小时前
【寒草呈献】当巴菲特走进 AI 投研助手
人工智能·架构