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
相关推荐
电子_咸鱼25 分钟前
动态规划经典题解：单词拆分（LeetCode 139）得贤招聘官5 小时前
精准招聘新纪元：AI 重构选才逻辑ChatPPT_YOO6 小时前
智能演示时代：8款免费AI PPT生成工具全面评测却道天凉_好个秋6 小时前
OpenCV(二十一)：图像的放大与缩小风暴之零7 小时前
卡尔曼滤波学习DatGuy7 小时前
Week 24: 深度学习补遗：Vision Transformer (ViT) 复现A尘埃8 小时前
项目三：信息抽取与图谱问答（医疗科研文献知识图谱与智能问答平台）风轻扬7778 小时前
SQLAlchemy2.0使用潘帕斯的雄鹰8 小时前
直观理解注意力机制鹿鸣悠悠8 小时前
AI测试（含大模型）与普通测试的区别及实施方法