1.cmd输入nvcc -V
2.得到 cuda版本后,去pytorch官网
3.根据自己的cuda进行选择
4.复制上述链接,进入cmd
5.cmd中输入activate XXX
,这里的"XXX"指代自己在工程中用到的环境
6.进入后,将刚才链接粘贴,回车等待下载结束
7.安装无报错后,回到train.py运行即可
如何从cpu改为gpu,pytorch,cuda
最新推荐文章于 2024-06-11 22:30:21 发布
1.cmd输入nvcc -V
2.得到 cuda版本后,去pytorch官网
3.根据自己的cuda进行选择
4.复制上述链接,进入cmd
5.cmd中输入activate XXX
,这里的"XXX"指代自己在工程中用到的环境
6.进入后,将刚才链接粘贴,回车等待下载结束
7.安装无报错后,回到train.py运行即可