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
相关推荐
心疼你的一切1 分钟前
昇腾CANN实战落地：从智慧城市到AIGC，解锁五大行业AI应用的算力密码AI绘画哇哒哒5 分钟前
【干货收藏】深度解析AI Agent框架：设计原理+主流选型+项目实操，一站式学习指南数据分析能量站7 分钟前
Clawdbot（现名Moltbot）-现状分析那个村的李富贵11 分钟前
CANN加速下的AIGC“即时翻译”：AI语音克隆与实时变声实战二十雨辰12 分钟前
[python]-AI大模型陈天伟教授12 分钟前
人工智能应用- 语言理解：04.大语言模型Luhui Dev12 分钟前
AI 与数学的融合：技术路径、应用前沿与未来展望（2026 版）Yvonne爱编码22 分钟前
JAVA数据结构 DAY6-栈和队列chian-ocean23 分钟前
量化加速实战：基于 `ops-transformer` 的 INT8 Transformer 推理那个村的李富贵23 分钟前
从CANN到Canvas：AI绘画加速实战与源码解析