选择Pytorch安装版本时,需要确认自己的计算平台是否支持GPU计算,适合哪个版本。 cmd 里输入nvidia-smi,即可查看本地CUDA版本 CMD界面.png 选择不高于该版本的计算平台 Pytorch版本选择界面.png