如何从cpu改为gpu,pytorch,cuda

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

你可能感兴趣的:(pytorch,人工智能,python)