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
相关推荐
GIS数据转换器1 小时前
地下空间综合管廊运营管理平台dozenyaoyida6 小时前
AI与大模型新闻日报 | 2026-10-17遇码7 小时前
流式调用中途断连别直接报错：瞬态故障识别、退避重试与中文根因诊断的搭建方法信誓旦旦的程序猿7 小时前
【Python量化实战宝典 #08】可转债与债券接口手册：转股价·溢价率·纯债价值（Python 可跑）救救孩子把7 小时前
AI面试八股文-21_21-多模态Agent面试题-Vision-Language-Agent砚底藏山河7 小时前
python量化：行情数据回溯今年下半年7 小时前
AI智能体项目实战记录（02）：初始化 Python 项目Jlzn88887 小时前
华东电芯分选通道数怎么配才不堵线Williamtym7 小时前
PickGo Lab实践：用浏览器、Python和MCP串起机器人搬运仿真