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
相关推荐
Bruce_Liuxiaowei3 分钟前
Python 字典默认值完全指南：从 KeyError 到 defaultdict乃嘿仔4 分钟前
AI 热点日报 · 2026-09-27数据杂坛9 分钟前
【Python程序开发系列】一文搞清楚Gunicorn以及和Nginx的区别weixin_1772972206910 分钟前
成立三年估值80亿美元：法律AI公司Harvey给中国企业的四个启发镭封10 分钟前
从人工到AI：短视频配音方式正在发生哪些变化初学大模型13 分钟前
先剔后识：OpenCV三维预处理与YOLO语义检测协同的机器人视觉系统可行性分析zjh90053014 分钟前
Visual C++ 2010 安装使用完整包：高手秘籍IT·陈寒19 分钟前
Vite热更新失效？你可能漏了这个配置项我想问问天20 分钟前
Jev 是做什么的？聊聊它能帮 LLM 分担哪些工作-cywen-28 分钟前
OpenSeg