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
相关推荐
yannan201903132 分钟前
【算法】（Python）动态规划埃菲尔铁塔_CV算法4 分钟前
人工智能图像算法：开启视觉新时代的钥匙EasyCVR4 分钟前
EHOME视频平台EasyCVR视频融合平台使用OBS进行RTMP推流，WebRTC播放出现抖动、卡顿如何解决？打羽毛球吗️11 分钟前
机器学习中的两种主要思路：数据驱动与模型驱动蒙娜丽宁12 分钟前
《Python OpenCV从菜鸟到高手》——零基础进阶，开启图像处理与计算机视觉的大门！光芒再现dev14 分钟前
已解决，部署GPTSoVITS报错‘AsyncRequest‘ object has no attribute ‘_json_response_data‘好喜欢吃红柚子28 分钟前
万字长文解读空间、通道注意力机制机制和超详细代码逐行分析（SE,CBAM,SGE,CA,ECA,TA）小馒头学python32 分钟前
机器学习是什么？AIGC又是什么？机器学习与AIGC未来科技的双引擎神奇夜光杯41 分钟前
Python酷库之旅-第三方库Pandas(202)正义的彬彬侠44 分钟前
《XGBoost算法的原理推导》12-14决策树复杂度的正则化项 公式解析