查看英伟达显卡对CUDA版本的支持

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


CMD界面.png

选择不高于该版本的计算平台


Pytorch版本选择界面.png

你可能感兴趣的:(查看英伟达显卡对CUDA版本的支持)