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