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
相关推荐
奈斯。zs16 分钟前
yjs08——矩阵、数组的运算Melody205016 分钟前
tensorflow-dataset 内网下载 指定目录学步_技术17 分钟前
Python编码系列—Python抽象工厂模式：构建复杂对象家族的蓝图DisonTangor1 小时前
阿里通义千问开源Qwen2.5系列模型：Qwen2-VL-72B媲美GPT-4豆浩宇1 小时前
Halcon OCR检测 免训练版Narutolxy1 小时前
Python 单元测试：深入理解与实战应用20240919LLSU131 小时前
聚星文社AI软件小说推文软件JackieZhengChina1 小时前
吴泳铭：AI最大的想象力不在手机屏幕，而是改变物理世界ShuQiHere1 小时前
【ShuQiHere】 探索数据挖掘的世界：从概念到应用嵌入式杂谈1 小时前
OpenCV计算机视觉：探索图片处理的多种操作