1.cmd输入nvcc -V
2.得到 cuda版本后，去pytorch官网
3.根据自己的cuda进行选择
4.复制上述链接，进入cmd
5.cmd中输入activate XXX,这里的"XXX"指代自己在工程中用到的环境
6.进入后，将刚才链接粘贴，回车等待下载结束
7.安装无报错后，回到train.py运行即可
如何从cpu改为gpu，pytorch，cuda
-electronic-engineer2023-08-15 12:20
相关推荐
水獭比特7 小时前
Vercel AI SDK Files V4：把文件交付做成可验收闭环迷迭香yy7 小时前
基金档案数据工程实战从收入分析到持仓穿透的Python解析 IG50免费开源股票数据API接口东方小月8 小时前
一篇文章带你深入拆解Skill的本质与工程实现，让你不再滥用Skill代码方舟8 小时前
零信任架构实战：基于天远车辆估值构建自动化二手车评估网关清水白石0088 小时前
Python 类型设计深度解析：TypedDict 能否替代 dataclass？从 JSON 数据边界到 API 设计的最佳实践xiaohaiAIgeo8 小时前
【2026年】实验室IoT三层部署架构详解zykk8 小时前
Codex 真的不再压缩上下文了吗？从 Compaction 到硬切窗口陈奕昆8 小时前
Wand-Enhancer 图像增强工具新手实战指南m4Rk_8 小时前
【论文阅读】Agent 记忆机制（57）：MemSearch-o1——从查询词元生长证据，重组 Deep Search 记忆路径