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
相关推荐
EntyIU几秒前
mineru从安装部署到测试使用完整指南汤姆yu1 分钟前
SkyClaw‑v1.0昆仑万维天工 AI如此这般英俊3 分钟前
手撕Claude Code—第一章 agent-loopC137的本贾尼6 分钟前
Spring AI Alibaba 模型全家桶：接入通义、百川、LLaMA 等第三方 LLM志栋智能12 分钟前
小步快跑：从单一场景开启超自动化巡检之旅lauo15 分钟前
从FunloomAI到ibbot：当你的手机不再是“手机”，而是你的AI副脑和生产节点实在智能RPA19 分钟前
AI Agent在制造业预测性维护上的算法精度怎样验证？深度拆解2026工业智能体实测表现我是大AI22 分钟前
搜极星 GEO：让 AI 精准推荐，品牌不再隐形明志数科28 分钟前
工业场景数据标注跟实验室标注有什么不同2601_9577875830 分钟前
企业内容矩阵系统：AI赋能下的全链路运营与获客升级