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
相关推荐
roman_日积跬步-终至千里2 小时前
AI 能让一个人成为团队吗？从数据中台到 OPC，重新理解软件工程能力在所不辞兄2 小时前
为什么神经网络在训练集损失已接近零后继续训练，测试性能仍可能继续提升，而不是单纯地越来越过拟合？AI砖家2 小时前
ChatGPT Images 2.6 正式支持 GIF 生成：一张图片也能动起来，AI 动图创作时代来了！希和20273 小时前
免费AI视频生成工具汇总（国内优先，适合做漫剧、短视频，Obsidian笔记版）水如烟3 小时前
孤能子视角：逆溯法(存稿)——从终点参照解往回拆必经关卡用户2991422196803 小时前
GitHub 热榜项目：周榜（2026-11-08）天天被压力3 小时前
【A股分钟级高频数据完全手册 #07】北交所历史分钟K线接口：京市Pro版分时批量用户2991422196804 小时前
AI 日报（2026年11月8日）天天被压力4 小时前
【A股分钟级高频数据完全手册 #08】股票日、分时数据接口对比：普通分时与Pro版差异实测miofly4 小时前
NVIDIA 开源 Cosmos 3 四款模型，用于交互式世界模拟与自动驾驶场景生成