如何从cpu改为gpu，pytorch，cuda_怎么将cpu换成gpu

作者：木道寻08 | 2024-07-19 12:13:46

踩

怎么将cpu换成gpu

1.cmd输入nvcc -V
2.得到 cuda版本后，去pytorch官网
在这里插入图片描述
3.根据自己的cuda进行选择

4.复制上述链接，进入cmd
5.cmd中输入activate XXX,这里的"XXX"指代自己在工程中用到的环境
6.进入后，将刚才链接粘贴，回车等待下载结束
7.安装无报错后，回到train.py运行即可

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/木道寻08/article/detail/851436