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
相关推荐
CITY_OF_MO_GY7 分钟前
Spark-TTS：基于大模型的文本语音合成工具阿丢是丢心心13 分钟前
【从0到1搞懂大模型】神经网络的实现：数据策略、模型调优与评估体系（3）新智元13 分钟前
10²⁶参数，AGI 还需 70 年！清华人大预测届时 GPU 总价达 4000 万倍苹果市值何大春21 分钟前
【对话推荐系统综述】Broadening the View: Demonstration-augmented Prompt Learning for CRWenGyyyL22 分钟前
使用OpenCV和MediaPipe库——增强现实特效（在手腕添加虚拟手表）东临碣石8224 分钟前
【英伟达AI论文】多模态大型语言模型的高效长视频理解我去热饭25 分钟前
【完整记录】基于腾讯云HAI+DeepSeek快速开发法律咨询(小律师)辅助平台过程CoovallyAIHub27 分钟前
一码难求的Manus，又对计算机视觉产生冲击？复刻开源版已在路上！是理不是里_28 分钟前
人工智能里的深度学习指的是什么？