1.首先查看自己电脑的CUDA
win+r,输入cmd
命令行输入nvidia-smi,查看自己CUDA的版本,我的是10.2,所以在安装pytorch时,版本不能选择高于10.2的。
2.打开anaconda prompt,输入conda create -n pytorch python=3.6,python的版本你可以自己选择,我这里选择的是与anaconda中的python版本相同的
输入y之后会出现如下界面:
输入conda activate pytorch代码,进入pytorch环境
3.进入pytorch官网,我们一般都会选择10.2的版本,conda进行安装,但是命令显示10.2的版本不再支持windows系统,不要慌,我们可以通过下方进入以前的版本进行安装
输入以下命令进行安装,时间可能会有点长
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
按照提示走一下
显示done之后输入pip list看是否有torch相关的包
然后通过以下操作验证是否成功