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
相关推荐
clorisqqq8 分钟前
人工智能现代方法笔记 第1章 绪论（1/2）kisshuan123968 分钟前
YOLO11-RepHGNetV2实现甘蔗田杂草与作物区域识别详解焦耳热科技前沿14 分钟前
北京科技大学/理化所ACS Nano：混合价态Cu₂Sb金属间化合物实现高效尿素电合成C+-C资深大佬20 分钟前
Creo 11.0 全功能解析：多体设计 + 仿真制造，机械设计效率翻倍下载安装浔川python社35 分钟前
【维护期间重要提醒】请勿使用浔川 AI 翻译 v6.0 翻译违规内容CS创新实验室1 小时前
AI 与编程min1811234561 小时前
深度伪造内容的检测与溯源技术!chen1 小时前
Error: error:0308010C:digital envelope routines::unsupporte_codemonster1 小时前
高斯卷积的可加性定理小北方城市网1 小时前
分布式锁实战指南：从选型到落地，避开 90% 的坑