一、获取设备上显卡型号 打开“设备管理器”查看显卡型号,我的显卡是RTX4060; 二、更新显卡驱动 打开NVIDIA官网,链接,找到对应的显卡驱动,默认下载最新的版本; 通过“win+r”打开cmd,输入nvidia-smi; nvidia-smi 显示显卡信息; 其中的CUDA Version:12.3指的是我的设备最高支持12.3的cuda版本; 三、选择需要安装的pytorch版本 打开pytorch官网选择需要安装的pytorch版本及其对应的cuda版本;