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
相关推荐
冬奇Lab6 小时前
Workflow 系列（04）：Multi-Agent 协调——编排器边界、并发控制与上下文隔离冬奇Lab6 小时前
每日一个开源项目（第147篇）：HyperGraphRAG - 用超图表示 N 元关系，RAG 的第三代范式甲维斯6 小时前
Github + 阿里云oss实现类似codex的自动更新！阿里云大数据AI技术8 小时前
光轮智能 × 阿里云：共建 Physical AI 云上数据、评测与持续学习基础设施机器之心8 小时前
实锤了：Claude Code偷查用户，时区、中国AI实验室全是关键词网易云信8 小时前
Cursor点燃个人开发者，企业级AI为何频频受挫？Agent工厂从提效工具到AI员工的跃迁网易云信8 小时前
解锁触手可及的温暖：网易智企 x Wander Puffs AI 云游泡芙转转技术团队8 小时前
从 PRD 到可验证代码：AI 需求开发闭环实践