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
相关推荐
小饕12 分钟前
AI编程的三阶段演化：哪些方向真正值得投入，哪些被高估了蔡俊锋24 分钟前
把1500个业务的大迁移，做成了可复用流水线用 Skill+Agent+Rule，省下 60 人年的实战复盘ZGi.ai25 分钟前
AI中台和AI工具的区别：为什么说前者是基础设施而后者是应用飘落的数码折腾日记31 分钟前
OpenClaw 是什么？让 AI 真正 “动手“ 帮你干活的秘密武器fthux37 分钟前
用了 GitZip 这么多年，我动手做了一个「Pro」版Zik----38 分钟前
DAEFR （ICLR 2024）— 盲脸超分模型解读TheRouter42 分钟前
Agent Harness系列（三）：记忆层的3种持久化架构——从SQLite到向量库一切皆是因缘际会1 小时前
从概率生成到内生心智：2026大模型瓶颈与下一代AI演进方向头发够用的程序员1 小时前
C++和Python面试经典算法汇总（一）X54先生（人文科技）1 小时前
《元创力》纪实录·心田记釉下新声：当《纪·念》成为可聆听的星轨