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
相关推荐
Guheyunyi2 小时前
智慧消防管理系统如何重塑安全未来ZZY_dl2 小时前
训练数据集(三)：真实场景下采集的课堂行为目标检测数据集，可直接用于YOLO各版本训练小鸡吃米…2 小时前
Python PyQt6教程七-控件yiersansiwu123d3 小时前
AI伦理治理：在创新与规范之间寻找动态平衡1916zz3 小时前
Extreme programing 方利喆 _ 江贤晟长安牧笛3 小时前
智能鞋柜—脚气终结者，内置温湿度传感器和紫外线灯，晚上回家，把鞋放进去，自动检测湿度，湿度超标就启动烘干+紫外线杀菌，第二天穿鞋干燥无异味。华清远见成都中心3 小时前
成都理工大学&华清远见成都中心实训,助力电商人才培养爱好读书3 小时前
AI生成er图/SQL生成er图在线工具CNRio3 小时前
智能影像：AI视频生成技术的战略布局与产业变革weixin_457760003 小时前
PIL库将图片位深度是1、8、32统一转换为24的方法