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
相关推荐
渡我白衣7 分钟前
未来的 AI 操作系统（二）——世界即界面：自然语言成为新的人机交互协议道之极万物灭8 分钟前
Python操作word实战墨利昂13 分钟前
词向量：自然语言处理技术体系的核心基石格林威24 分钟前
可见光工业相机半导体制造领域中的应用星期天要睡觉31 分钟前
计算机视觉（opencv）——基于 MediaPipe 人体姿态检测Python私教42 分钟前
DRF：Django REST Framework框架介绍资讯全球1 小时前
2025机器人自动化打磨抛光设备及汽车零件打磨新技术10月应用解析数智前线1 小时前
京东零售的AI野心：为每个商家打造自己的“AI战队”Cl_rown去掉l变成C1 小时前
第N7周打卡：调用Gensim库训练Word2Vec模型腾讯云开发者2 小时前
腾讯云TVP走进美的，共探智能制造新范式