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
相关推荐
少废话h2 小时前
解决Flink中ApacheCommonsCLI版本冲突serve the people2 小时前
TensorFlow 图执行（tf.function）的 “非严格执行（Non-strict Execution）” 特性天命码喽c2 小时前
GraphRAG-2.7.0整合Milvus-2.5.1泰迪智能科技2 小时前
图书推荐分享 | 堪称教材天花板，深度学习教材-TensorFlow 2 深度学习实战（第2版）（微课版）吴佳浩4 小时前
LangChain 深入网安-轩逸7 小时前
回归测试原则：确保软件质量的基石Mr_Xuhhh7 小时前
YAML相关LplLpl117 小时前
AI 算法竞赛通关指南：基于深度学习的图像分类模型优化实战咖啡の猫7 小时前
Python中的变量与数据类型依米s7 小时前
各年度人工智能大会WAIC核心议题（持续更新）