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
相关推荐
2201_754918411 小时前
OpenCV 背景建模详解：从原理到实战CopyLower1 小时前
苹果计划将AI搜索集成至Safari：谷歌搜索下降引发的市场变革wd2099881 小时前
2025年Ai写PPT工具推荐，这5款Ai工具可以一键生成专业PPT张飞飞飞飞飞1 小时前
语音识别——声纹识别archko2 小时前
语音识别-3,添加ai问答Bowen_CV4 小时前
AI 赋能防艾宣传：从创意到实践，我的 IP 形象设计之旅你是一个铁憨憨4 小时前
使用深度学习预训练模型检测物体reasonsummer4 小时前
【办公类-100-01】20250515手机导出教学照片，自动上传csdn+最大化、最小化Vs界面Doker 多克5 小时前
Python-Django系列—日志AI算法工程师Moxi5 小时前
什么时候可以开始学习深度学习？