安装pytorch在其官网中需要对应Compute Platform的CUDA版本进行安装。查看windows系统的CUDA如下：
键盘win+R 输入cmd打开控制台，输入如下指令即可查看版本。
nvidia-smi
然后在pytorch官网选择合适的版本进行下载即可。
windows查看CUDA版本
安装pytorch在其官网中需要对应Compute Platform的CUDA版本进行安装。查看windows系统的CUDA如下：
键盘win+R 输入cmd打开控制台，输入如下指令即可查看版本。
nvidia-smi
然后在pytorch官网选择合适的版本进行下载即可。