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
相关推荐
网易云信几秒前
制造业、零售与物流的“神经末梢”：为什么企业级IM是这些行业的数字基建？千瓜6 分钟前
用户洞察：负鼠走红？解读新世代“动物人格”中微极客18 分钟前
2026年生产级RAG技术栈选型：LangChain+Cohere Rerank实战ai_coder_ai23 分钟前
在自动化脚本中如何实现播音？硅徒24 分钟前
红蓝对抗实战复盘：一场攻防演练里暴露的真实短板ksueh29 分钟前
什么AI工具可以写小说？2026实测10款工具横评：从免费通用到专业工作台，一篇说透ellenwan202630 分钟前
近期AI协作量化实现，先补规则清晰度和流程完整性陆枫Larry31 分钟前
为什么 AI 平台选择「月租套餐 + 点数计费」双轨模式？李杰SEO35 分钟前
2026企业品牌：借GEO优化抢AI搜索全域流量卷无止境35 分钟前
Python 类型注解与运行时反射：从原理到工程实践