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
相关推荐
风象南34 分钟前
最适合新手先装的 20 个 OpenClaw Skills 来了！明月_清风1 小时前
Python 内存手术刀：sys.getrefcount 与引用计数的生死时速明月_清风1 小时前
Python 消失的内存：为什么 list=[] 是新手最容易踩的“毒苹果”？小兵张健11 小时前
35岁程序员的春天来了大怪v12 小时前
AI抢饭？前端佬：我要验牌！冬奇Lab12 小时前
OpenClaw 深度解析（六）：节点、Canvas 与子 Agent刀法如飞13 小时前
AI提示词框架深度对比分析IT_陈寒15 小时前
Python开发者必知的5大性能陷阱：90%的人都踩过的坑！1G15 小时前
openclaw控制浏览器/自动化的playwright MCP + Mcporter方案实现踩着两条虫15 小时前
VTJ.PRO 双向代码转换原理揭秘